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 19:43:25 -04:00
parent 8df13035eb
commit cf57be9dca
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+2
View File
@@ -63,6 +63,7 @@ class ExampleModel(RootModule):
:param data_batch:
:return:
"""
pdb.set_trace()
# forward pass
x, y = data_batch
x = x.view(x.size(0), -1)
@@ -80,6 +81,7 @@ class ExampleModel(RootModule):
:param data_batch:
:return:
"""
pdb.set_trace()
x, y = data_batch
x = x.view(x.size(0), -1)
y_hat = self.forward(x)