updated args

This commit is contained in:
William Falcon committed 2019-06-25 19:17:17 -04:00
1 parent c4da914747
commit 89410e9090
2 files changed
+8 -5

No files matched your search

+1
View File
@@ -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)