mirror of https://github.com/coqui-ai/TTS.git
differentiate modules on TB with a prefix
parent
e71a377202
commit
5116a20c86
2
train.py
2
train.py
|
@ -667,7 +667,7 @@ if __name__ == '__main__':
|
|||
os.chmod(OUT_PATH, 0o775)
|
||||
|
||||
LOG_DIR = OUT_PATH
|
||||
tb_logger = TensorboardLogger(LOG_DIR)
|
||||
tb_logger = TensorboardLogger(LOG_DIR, model_name='TTS')
|
||||
|
||||
# write model desc to tensorboard
|
||||
tb_logger.tb_add_text('model-description', c['run_description'], 0)
|
||||
|
|
Loading…
Reference in New Issue