wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix root node resolution (#1954)

Browse Source
This commit is contained in:
Loïc Grobol
2020-05-27 22:50:37 -04:00
committed by GitHub
parent df78e84060
commit c3cf33d1de
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/distrib_data_parallel.py
+2 -2
View File
@@ -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]