Fix root node resolution (#1954)

Lightning-AI · Jun 29, 2020 · 7867466 · 7867466
1 parent 6b3b318
commit 7867466
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/distrib_data_parallel.py b/pytorch_lightning/trainer/distrib_data_parallel.py
@@ -426,8 +426,8 @@ def load_spawn_weights(self, original_model):
 
     def resolve_root_node_address(self, root_node):
         if '[' in root_node:
-            name = root_node.split('[')[0]
-            number = root_node.split(',')[0]
+            name, numbers = root_node.split('[', maxsplit=1)
+            number = numbers.split(',', maxsplit=1)[0]
             if '-' in number:
                 number = number.split('-')[0]