(self, batch, batch_idx)
| 128 | return (loss,) |
| 129 | |
| 130 | def training_step(self, batch, batch_idx) -> Dict: |
| 131 | loss_tensors = self._step(batch) |
| 132 | logs = {name: loss for name, loss in zip(self.loss_names, loss_tensors)} |
| 133 | return {"loss": loss_tensors[0], "log": logs} |
| 134 | |
| 135 | def validation_step(self, batch, batch_idx) -> Dict: |
| 136 | return self._generative_step(batch) |