From fe05f629848e63adc6f1c663d99d4e95b214e2d9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 3 Aug 2019 21:48:49 -0500 Subject: [PATCH] fixed drop prob import --- .../examples/new_project_templates/lightning_module_template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py index ca8f604b23..993a7b77da 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/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)