Method training_step

trainer.py:602–603

(self, batch, batch_idx)

        return loss

    def training_step(self, batch, batch_idx):
        return self.run_step(batch, batch_idx)

    def on_save_checkpoint(self, checkpoint):
        state = {}