From d9f1268f99f3a91078f9f4806e9283b615c51c03 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Mon, 31 May 2021 15:47:07 +0200
Subject: [PATCH] init tb_logger None for rank > 0 processes

---
 TTS/utils/arguments.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py
index 1b5a424b..5e6acd1d 100644
--- a/TTS/utils/arguments.py
+++ b/TTS/utils/arguments.py
@@ -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 = {}