mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
testing master_Addr flag
This commit is contained in:
@@ -384,8 +384,8 @@ class Trainer(TrainerIO):
|
||||
:return:
|
||||
"""
|
||||
# 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'
|
||||
|
||||
# where to store ip_table
|
||||
ip_file_dir = os.path.join(self.cluster.log_path, 'ip_tables')
|
||||
@@ -409,6 +409,9 @@ class Trainer(TrainerIO):
|
||||
|
||||
return root_ip
|
||||
else:
|
||||
# sleep 10 seconds first to give file chance to write
|
||||
sleep(10)
|
||||
|
||||
# wait up to 120 seconds until proc 0 writes
|
||||
# once written, read proc 0's address and use it to configure server
|
||||
for i in range(0, 120):
|
||||
|
||||
Reference in New Issue
Block a user