[ci skip] implement #418

pull/423/head
Eren Gölge 2021-04-06 16:24:13 +02:00
parent f46a275b22
commit 9782d9ea5d
1 changed files with 3 additions and 2 deletions

View File

@ -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