mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
updated args
This commit is contained in:
@@ -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