init tb_logger None for rank > 0 processes

2026-09-20 12:30:22 +08:00 · 2021-05-31 15:47:07 +02:00
parent 301c516abd
commit d9f1268f99
1 changed files with 1 additions and 0 deletions
@@ -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 = {}