(self)
| 497 | return ret_tuple |
| 498 | |
| 499 | def run(self): |
| 500 | assert self.mode in ["train", "train_only", "val"] |
| 501 | if self.mode == "train": |
| 502 | if self.epoch > 0: |
| 503 | logging.info(f"Resuming training from epoch: {self.epoch}") |
| 504 | # resuming from a checkpoint |
| 505 | if self.is_intermediate_val_epoch(self.epoch - 1): |
| 506 | logging.info("Running previous val epoch") |
| 507 | self.epoch -= 1 |
| 508 | self.run_val() |
| 509 | self.epoch += 1 |
| 510 | self.run_train() |
| 511 | self.run_val() |
| 512 | elif self.mode == "val": |
| 513 | self.run_val() |
| 514 | elif self.mode == "train_only": |
| 515 | self.run_train() |
| 516 | |
| 517 | def _setup_dataloaders(self): |
| 518 | self.train_dataset = None |
no test coverage detected