[ci skip] implement #418

coqui-ai · Apr 6, 2021 · 9782d9e · 9782d9e
1 parent f46a275
commit 9782d9e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py
@@ -6,6 +6,7 @@
 import glob
 import os
 import re
+import json
 
 import torch
 from TTS.tts.utils.text.symbols import parse_symbols
@@ -190,7 +191,7 @@ def process_args(args, model_class):
 
  tb_logger = TensorboardLogger(log_path, model_name=model_class.upper())
 
- # write model desc to tensorboard
- tb_logger.tb_add_text("model-description", c["run_description"], 0)
+ # write model config to tensorboard
+ tb_logger.tb_add_text("model-config", f"<pre>{json.dumps(c, indent=4)}</pre>", 0)
 
  return c, out_path, audio_path, c_logger, tb_logger