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

updated args

Browse Source
This commit is contained in:
William Falcon
2019-06-25 19:17:17 -04:00
parent c4da914747
commit 89410e9090
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+1
View File
@@ -41,6 +41,7 @@ class ExampleModel(RootModule):
# TRAINING
# ---------------------
def forward(self, x):
x = self.c_d1(x)
x = F.tanh(x)
x = self.c_d1_bn(x)