From b58ec7ad5a1ed9dcd91bfd7f4ee4c8280e79fa5b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 26 Jun 2019 18:26:08 -0400 Subject: [PATCH] removed self.model refs --- 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 7dacf6ea..f141911e 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -201,8 +201,8 @@ class ExampleModel(RootModule): # network params parser.opt_list('--drop_prob', default=0.2, options=[0.2, 0.5], type=float, tunable=False) parser.add_argument('--in_features', default=28*28) - parser.add_argument('--hidden_dim', default=50000) parser.add_argument('--out_features', default=10) + parser.add_argument('--hidden_dim', default=50000) # use 500 for CPU, 50000 for GPU to see speed difference # data parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str)