Merge pull request #31 from williamFalcon/examples

fixed drop prob import

pytorch_lightning/examples/new_project_templates/lightning_module_template.py

@@ -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)