Skip to content

Commit 68d3ca0

Browse files
committed
Add barrier
1 parent e08a759 commit 68d3ca0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

algorithmic_efficiency/logger_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,8 @@ def get_log_dir(
7474
if resume.lower() != 'y':
7575
sys.exit()
7676

77+
if USE_PYTORCH_DDP:
78+
dist.barrier()
7779
logging.info(f'Creating experiment directory at {experiment_path}.')
7880
makedir(experiment_path)
7981
return experiment_path

0 commit comments

Comments
 (0)