From c3cf33d1de7de3444e3a4a7d45bc613d8e01a7c0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Grobol?= Date: Thu, 28 May 2020 02:50:37 +0000 Subject: [PATCH] Fix root node resolution (#1954) --- pytorch_lightning/trainer/distrib_data_parallel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/distrib_data_parallel.py b/pytorch_lightning/trainer/distrib_data_parallel.py index 7fea36ec..21bce343 100644 --- a/pytorch_lightning/trainer/distrib_data_parallel.py +++ b/pytorch_lightning/trainer/distrib_data_parallel.py @@ -426,8 +426,8 @@ class TrainerDDPMixin(ABC): 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]