(self, batch, batch_idx)
| 665 | return self.step(batch, l1_loss, "test") |
| 666 | |
| 667 | def test_step(self, batch, batch_idx): |
| 668 | return self.step(batch, l1_loss, "test") |
| 669 | |
| 670 | def step(self, batch, loss_fn, stage): |
| 671 | with torch.set_grad_enabled(stage == "train" or self.hparams.derivative): |