(self, batch, batch_idx)
| 133 | return {"loss": loss_tensors[0], "log": logs} |
| 134 | |
| 135 | def validation_step(self, batch, batch_idx) -> Dict: |
| 136 | return self._generative_step(batch) |
| 137 | |
| 138 | def validation_epoch_end(self, outputs, prefix="val") -> Dict: |
| 139 | self.step_count += 1 |