| 1232 | return self._run |
| 1233 | |
| 1234 | def on_train_begin(self, logs: _CallbackLogs | None = None) -> None: |
| 1235 | self.epochs = self.params['epochs'] |
| 1236 | assert self.epochs |
| 1237 | self.train_metrics = self.params['metrics'] |
| 1238 | assert self.train_metrics |
| 1239 | self._is_fit = True |
| 1240 | self.train_step = 0 |
| 1241 | |
| 1242 | if self._is_write(): |
| 1243 | self.run.define_metric("train/step") |
| 1244 | self.run.define_metric("train/*", step_metric="train/step") |
| 1245 | |
| 1246 | self.run.define_metric("epoch") |
| 1247 | self.run.define_metric("eval/*", step_metric="epoch") |
| 1248 | |
| 1249 | def on_epoch_begin( |
| 1250 | self, epoch: int, logs: _CallbackLogs | None = None |