From e325ba5f6842436e135b1e420224b341b873c106 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 10:57:34 -0400 Subject: [PATCH] testing slurm ddp --- examples/new_project_templates/lightning_module_template.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 9968a12de5..b1ea9d9b10 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -210,9 +210,9 @@ class LightningTemplateModel(LightningModule): # 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('--out_features', default=10) - parser.add_argument('--hidden_dim', default=50000) # use 500 for CPU, 50000 for GPU to see speed difference + 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 # data parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str)