updated args

2019-06-25 19:56:47 -04:00 · 2019-06-25 19:56:47 -04:00 · 69be732b11
parent b59af1813b
commit 69be732b11
1 changed files with 1 additions and 1 deletions
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@ -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)