(self, batch, batch_idx)
| 653 | return self.model(data) |
| 654 | |
| 655 | def training_step(self, batch, batch_idx): |
| 656 | loss_fn = mse_loss if self.hparams.loss_type == 'MSE' else l1_loss |
| 657 | |
| 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): |