(self, max_epochs: int)
| 41 | print(f"Epoch {epoch} | Training checkpoint saved at {PATH}") |
| 42 | |
| 43 | def train(self, max_epochs: int): |
| 44 | for epoch in range(max_epochs): |
| 45 | self._run_epoch(epoch) |
| 46 | if epoch % self.save_every == 0: |
| 47 | self._save_checkpoint(epoch) |
| 48 | |
| 49 | |
| 50 | def load_train_objs(): |
no test coverage detected