wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +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
pl_examples/basic_examples/lightning_module_template.py
+2 -2
View File
@@ -106,7 +106,7 @@ class LightningTemplateModel(LightningModule):
x, y = batch
x = x.view(x.size(0), -1)
y_hat = self.forward(x)
y_hat = self(x)
# calculate loss
loss_val = self.loss(y, y_hat)
@@ -133,7 +133,7 @@ class LightningTemplateModel(LightningModule):
"""
x, y = batch
x = x.view(x.size(0), -1)
y_hat = self.forward(x)
y_hat = self(x)
loss_val = self.loss(y, y_hat)