(self, batch, batch_idx)
| 49 | return loss |
| 50 | |
| 51 | def test_step(self, batch, batch_idx): |
| 52 | return self.common_step(batch, batch_idx) |
| 53 | |
| 54 | def configure_optimizers(self): |
| 55 | optimizer = optim.SGD(self.model.parameters(), self.lr, |
nothing calls this directly
no test coverage detected