mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
updated args
This commit is contained in:
1 parent
b59af1813b
commit
69be732b11
1 file changed
+1
-1
@@ -40,7 +40,7 @@ class ExampleModel(RootModule):
|
||||
# ---------------------
|
||||
# TRAINING
|
||||
# ---------------------
|
||||
def forward(self, x, a):
|
||||
def forward(self, x):
|
||||
|
||||
x = self.c_d1(x)
|
||||
x = F.tanh(x)
|
||||
|
||||
Reference in new issue
Block a user