From 9782d9ea5de381d88eda7b1c1c362025b61cd991 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 6 Apr 2021 16:24:13 +0200
Subject: [PATCH] [ci skip] implement #418

---
 TTS/utils/arguments.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py
index 053993ca..17a76aa6 100644
--- a/TTS/utils/arguments.py
+++ b/TTS/utils/arguments.py
@@ -6,6 +6,7 @@ import argparse
 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