wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-26 13:00:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #31 from williamFalcon/examples

Browse Source 
fixed drop prob import
This commit is contained in:
William Falcon authored and GitHub committed 2019-08-03 22:03:13 -05:00
2 parents 085195e09b fe05f62984
commit ea7c8e55d7
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/examples/new_project_templates/lightning_module_template.py
+1
View File
@@ -224,6 +224,7 @@ class LightningTemplateModel(LightningModule):
parser.add_argument('--in_features', default=28*28, type=int)
parser.add_argument('--out_features', default=10, type=int)
parser.add_argument('--hidden_dim', default=50000, type=int) # use 500 for CPU, 50000 for GPU to see speed difference
parser.opt_list('--drop_prob', default=0.2, options=[0.2, 0.5], type=float, tunable=False)
# data
parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str)