init tb_logger None for rank > 0 processes

This commit is contained in:
Eren Gölge 2021-05-31 15:47:07 +02:00
parent 301c516abd
commit d9f1268f99
1 changed files with 1 additions and 0 deletions

View File

@ -152,6 +152,7 @@ def process_args(args):
experiment_path = create_experiment_folder(config.output_path, config.run_name, args.debug)
audio_path = os.path.join(experiment_path, "test_audios")
# setup rank 0 process in distributed training
tb_logger = None
if args.rank == 0:
os.makedirs(audio_path, exist_ok=True)
new_fields = {}