in training/trainer.py [0:0]
def run(self):
assert self.mode in ["train", "train_only", "val"]
if self.mode == "train":
if self.epoch > 0:
logging.info(f"Resuming training from epoch: {self.epoch}")
# resuming from a checkpoint
if self.is_intermediate_val_epoch(self.epoch - 1):
logging.info("Running previous val epoch")
self.epoch -= 1
self.run_val()
self.epoch += 1
self.run_train()
self.run_val()
elif self.mode == "val":
self.run_val()
elif self.mode == "train_only":
self.run_train()