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]