fix gpu template (#2255)

Lightning-AI · Jun 19, 2020 · 4885cfa · 4885cfa
1 parent b5a2f1e
commit 4885cfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pl_examples/basic_examples/gpu_template.py b/pl_examples/basic_examples/gpu_template.py
@@ -23,7 +23,7 @@ def main(hparams):
     # ------------------------
     # 1 INIT LIGHTNING MODEL
     # ------------------------
-    model = LightningTemplateModel(hparams)
+    model = LightningTemplateModel(**vars(hparams))
 
     # ------------------------
     # 2 INIT TRAINER
@@ -61,7 +61,7 @@ def main(hparams):
         '--distributed_backend',
         type=str,
         default='dp',
-        help='supports three options dp, ddp, ddp2'
+        help='supports four options dp, ddp, ddp2, ddp_spawn'
     )
     parent_parser.add_argument(
         '--use_16bit',