mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
updated args
This commit is contained in:
1 parent
c4da914747
commit
89410e9090
2 files changed
+8
-5
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user