From 69be732b116de61041853063340961fad2d7be30 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 19:56:47 -0400 Subject: [PATCH] updated args --- docs/source/examples/example_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py index 6570c5d31e..ddd0308379 100644 --- 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)