wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated docs

Browse Source
This commit is contained in:
William Falcon
2019-07-25 11:58:06 -04:00
parent d272f29c88
commit 4562580461
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4
View File
@@ -68,6 +68,10 @@ class CoolModel(ptl.LightningModule):
y_hat = self.forward(x) y_hat = self.forward(x)
return {'val_loss': self.my_loss(y_hat, y)} 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): def configure_optimizers(self):
return [torch.optim.Adam(self.parameters(), lr=0.02)] return [torch.optim.Adam(self.parameters(), lr=0.02)]
docs/LightningModule/RequiredTrainerInterface.md
+4
View File
@@ -57,6 +57,10 @@ class CoolModel(ptl.LightningModule):
y_hat = self.forward(x) y_hat = self.forward(x)
return {'val_loss': self.my_loss(y_hat, y)} 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): def configure_optimizers(self):
return [torch.optim.Adam(self.parameters(), lr=0.02)] return [torch.optim.Adam(self.parameters(), lr=0.02)]