Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add explicit barrier after init dist groups

Browse Source
This commit is contained in:
Yin Li 2020-04-21 13:38:02 -04:00
parent 28d7cc9e9b
commit c9f468c568
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
map2map/train.py
View File

@ -542,6 +542,7 @@ def dist_init(rank, args):
world_size=args.world_size, world_size=args.world_size,
rank=rank, rank=rank,
) )
dist.barrier()
if rank == 0: if rank == 0:
os.remove(dist_file) os.remove(dist_file)