mirror of https://github.com/coqui-ai/TTS.git
[ci skip] implement #418
This commit is contained in:
parent
f46a275b22
commit
9782d9ea5d
|
@ -6,6 +6,7 @@ import argparse
|
||||||
import glob
|
import glob
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
|
import json
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
from TTS.tts.utils.text.symbols import parse_symbols
|
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())
|
tb_logger = TensorboardLogger(log_path, model_name=model_class.upper())
|
||||||
|
|
||||||
# write model desc to tensorboard
|
# write model config to tensorboard
|
||||||
tb_logger.tb_add_text("model-description", c["run_description"], 0)
|
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
|
return c, out_path, audio_path, c_logger, tb_logger
|
||||||
|
|
Loading…
Reference in New Issue