diff --git a/speechbrain/lobes/augment.py b/speechbrain/lobes/augment.py index e5f6e26786..95ac121685 100644 --- a/speechbrain/lobes/augment.py +++ b/speechbrain/lobes/augment.py @@ -504,7 +504,7 @@ def _prepare_csv(folder, filelist, csv_file, max_length=None): """ try: # make sure all processing reached here before main preocess create csv_file - sb.distributed.ddp_barrier() + sb.utils.distributed.ddp_barrier() if sb.utils.distributed.if_main_process(): with open(csv_file, "w") as w: w.write("ID,duration,wav,wav_format,wav_opts\n\n") diff --git a/speechbrain/utils/data_utils.py b/speechbrain/utils/data_utils.py index e28b2e5fa4..b8a343558b 100644 --- a/speechbrain/utils/data_utils.py +++ b/speechbrain/utils/data_utils.py @@ -300,7 +300,7 @@ def download_file( """ try: # make sure all processing reached here before main preocess create dest_dir - sb.distributed.ddp_barrier() + sb.utils.distributed.ddp_barrier() if sb.utils.distributed.if_main_process(): class DownloadProgressBar(tqdm.tqdm):