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 args

Browse Source
This commit is contained in:
William Falcon
2019-06-25 20:09:40 -04:00
parent bac0ef2d44
commit 2ac5cce67a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+2
View File
@@ -65,6 +65,7 @@ class ExampleModel(RootModule):
"""
# forward pass
x, y = data_batch
print(x.shape)
x = x.view(x.size(0), -1)
y_hat = self.forward(x)
@@ -86,6 +87,7 @@ class ExampleModel(RootModule):
:return:
"""
x, y = data_batch
print(x.shape)
x = x.view(x.size(0), -1)
y_hat = self.forward(x)