mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
Fix root node resolution (#1954)
This commit is contained in:
1 parent
df78e84060
commit
c3cf33d1de
1 file changed
+2
-2
@@ -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]
|
||||
|
||||
|
||||
Reference in new issue
Block a user