updated docs

Lightning-AI · Jul 25, 2019 · 4562580 · 4562580
1 parent d272f29
commit 4562580
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -68,6 +68,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)]
 

diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md
@@ -57,6 +57,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)]