Deployed 4562580 with MkDocs version: 1.0.4

2026-09-09 11:32:07 +08:00 · 2019-07-25 10:58:14 -05:00
parent f3a4149d50
commit 4c60f2e8c3
3 changed files with 5 additions and 1 deletions
@@ -944,6 +944,10 @@ class CoolModel(ptl.LightningModule):
        y_hat = self.forward(x)
        return {'val_loss': self.my_loss(y_hat, y)}

+    def validation_end(self, outputs):
+        avg_loss = torch.stack([x for x in outputs['val_loss']]).mean()
+        return avg_loss
+
    def configure_optimizers(self):
        return [torch.optim.Adam(self.parameters(), lr=0.02)]