(
self,
epoch: int,
bidx: int,
batch: T.Any,
)
| 1284 | setattr(self, name, None) |
| 1285 | |
| 1286 | def train_step( |
| 1287 | self, |
| 1288 | epoch: int, |
| 1289 | bidx: int, |
| 1290 | batch: T.Any, |
| 1291 | ): |
| 1292 | |
| 1293 | self.reset_losses() |
| 1294 | self.reset_outputs() |
| 1295 | |
| 1296 | all_loss_dict = self._step( |
| 1297 | epoch=epoch, |
| 1298 | bidx=bidx, |
| 1299 | batch=batch, |
| 1300 | update=True, |
| 1301 | ) |
| 1302 | return all_loss_dict |
| 1303 | |
| 1304 | def validation_step( |
| 1305 | self, |
nothing calls this directly
no test coverage detected