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