(self, max_epochs: int)
| 69 | print(f"Epoch {epoch} | Training snapshot saved at {self.snapshot_path}") |
| 70 | |
| 71 | def train(self, max_epochs: int): |
| 72 | for epoch in range(self.epochs_run, max_epochs): |
| 73 | self._run_epoch(epoch) |
| 74 | if self.global_rank == 0 and epoch % self.save_every == 0: |
| 75 | self._save_snapshot(epoch) |
| 76 | |
| 77 | |
| 78 | def load_train_objs(): |
no test coverage detected