(self, batch, batch_idx, *args)
| 658 | return self.step(batch, loss_fn, "train") |
| 659 | |
| 660 | def validation_step(self, batch, batch_idx, *args): |
| 661 | if len(args) == 0 or (len(args) > 0 and args[0] == 0): |
| 662 | # validation step |
| 663 | return self.step(batch, mse_loss, "val") |
| 664 | # test step |
| 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") |