updated args

This commit is contained in:
William Falcon committed 2019-06-25 19:56:47 -04:00
1 parent b59af1813b
commit 69be732b11
1 file changed
+1 -1
+1 -1
View File
@@ -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)