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

cleaning up demos

Browse Source
This commit is contained in:
William Falcon
2019-10-05 16:09:48 -04:00
parent 5c99da5861
commit a00e24ed80
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+2 -1
View File
@@ -919,7 +919,7 @@ class Trainer(TrainerIO):
# if user gave a port number, use that one instead
try:
port = os.environ['MASTER_PORT']
default_port = os.environ['MASTER_PORT']
except Exception:
os.environ['MASTER_PORT'] = str(default_port)
@@ -931,6 +931,7 @@ class Trainer(TrainerIO):
root_node = self.resolve_root_node_address(root_node)
os.environ['MASTER_ADDR'] = root_node
print(default_port)
dist.init_process_group("nccl", rank=self.proc_rank, world_size=self.world_size)