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

calling self.forward() -> self() (#1211)

Browse Source 
* self.forward() -> self()

* update changelog

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
This commit is contained in:
Jeremy Jordan
2020-03-27 08:17:56 +01:00
committed by GitHub
co-authored by Jirka Borovec
parent 2a4cd479e2
commit d394b80ac8
15 changed files with 46 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/base/debug.py
+2 -2
View File
@@ -26,12 +26,12 @@ class CoolModel(pl.LightningModule):
def training_step(self, batch, batch_idx):
x, y = batch
y_hat = self.forward(x)
y_hat = self(x)
return {'training_loss': self.my_loss(y_hat, y)}
def validation_step(self, batch, batch_idx):
x, y = batch
y_hat = self.forward(x)
y_hat = self(x)
return {'val_loss': self.my_loss(y_hat, y)}
def validation_epoch_end(self, outputs):