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

easy import for lightningModule

Browse Source
This commit is contained in:
William Falcon
2019-07-08 09:33:58 -04:00
parent 1e57a75ff9
commit 5fbe00837e
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+2 -3
View File
@@ -324,12 +324,11 @@ class Trainer(TrainerIO):
def __get_root_node_ip(self, proc_rank, nb_gpu_nodes, ip_file_dir):
# on one node we use localhost
# if nb_gpu_nodes == 1:
# return '127.0.0.1'
if nb_gpu_nodes == 1:
return '127.0.0.1'
# on multi-node, every node rank > 0 waits until rank 0
# saves the ip to disk
print(ip_file_dir)
ip_file = os.path.join(ip_file_dir, '.ip_meta')
if proc_rank == 0:
# get the proc 0 IP