updated args

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