[examples] bump pl=0.9.0 (#7053)

This commit is contained in:
Sam Shleifer
2020-10-11 16:39:38 -04:00
committed by GitHub
parent ba4bbd92bc
commit 827c519494
7 changed files with 27 additions and 42 deletions

View File

@@ -181,6 +181,7 @@ class SummarizationModule(BaseTransformer):
return self._generative_step(batch)
def validation_epoch_end(self, outputs, prefix="val") -> Dict:
self.step_count += 1
losses = {k: torch.stack([x[k] for x in outputs]).mean() for k in self.loss_names}
loss = losses["loss"]