(self, batch, batch_idx, dataloader_idx)
| 67 | update_module_step(self.model, self.current_epoch, self.global_step) |
| 68 | |
| 69 | def on_predict_batch_start(self, batch, batch_idx, dataloader_idx): |
| 70 | self.dataset = self.trainer.datamodule.predict_dataloader().dataset |
| 71 | self.preprocess_data(batch, 'predict') |
| 72 | update_module_step(self.model, self.current_epoch, self.global_step) |
| 73 | |
| 74 | def training_step(self, batch, batch_idx): |
| 75 | raise NotImplementedError |
nothing calls this directly
no test coverage detected