(self, max_epochs: int)
| 60 | print(f"Epoch {epoch} | Training checkpoint saved at {PATH}") |
| 61 | |
| 62 | def train(self, max_epochs: int): |
| 63 | for epoch in range(max_epochs): |
| 64 | self._run_epoch(epoch) |
| 65 | if self.gpu_id == 0 and epoch % self.save_every == 0: |
| 66 | self._save_checkpoint(epoch) |
| 67 | |
| 68 | |
| 69 | def load_train_objs(): |
no test coverage detected