wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-25 12:50:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated args

Browse Source
This commit is contained in:
William Falcon committed 2019-06-25 19:56:12 -04:00
1 parent 7814b2d449
commit b59af1813b
2 files changed
+1 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/source/examples/example_model.py
-3
View File
@@ -42,7 +42,6 @@ class ExampleModel(RootModule):
# ---------------------
def forward(self, x, a):
pdb.set_trace()
x = self.c_d1(x)
x = F.tanh(x)
x = self.c_d1_bn(x)
@@ -63,7 +62,6 @@ class ExampleModel(RootModule):
:param data_batch:
:return:
"""
pdb.set_trace()
# forward pass
x, y = data_batch
x = x.view(x.size(0), -1)
@@ -81,7 +79,6 @@ 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)