mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-25 12:50:57 +08:00
removed self.model refs
This commit is contained in:
1 parent
8f9672603b
commit
42fe76f794
1 file changed
+1
-1
@@ -201,7 +201,7 @@ 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=500)
|
||||
parser.add_argument('--hidden_dim', default=50000)
|
||||
parser.add_argument('--out_features', default=10)
|
||||
|
||||
# data
|
||||
|
||||
Reference in new issue
Block a user