removed self.model refs

2019-06-26 18:26:08 -04:00 · 2019-06-26 18:26:08 -04:00 · b58ec7ad5a
parent 301a4992f4
commit b58ec7ad5a
1 changed files with 1 additions and 1 deletions
--- 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)