mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
added multi-node locked ip search
This commit is contained in:
1 parent
6462cab351
commit
f4ab46e1c9
1 file changed
+1
@@ -383,6 +383,7 @@ class Trainer(TrainerIO):
|
||||
# now everyone waits until the file has world_size entries
|
||||
for i in range(0, 120):
|
||||
sleep(1.0)
|
||||
print('sleeping...')
|
||||
if os.path.exists(ip_file):
|
||||
lines = list(open(file=ip_file, mode='r'))
|
||||
if len(lines) == world_size:
|
||||
|
||||
Reference in new issue
Block a user