differentiate modules on TB with a prefix

pull/1/head
erogol 2020-05-30 18:07:01 +02:00
parent e71a377202
commit 5116a20c86
1 changed files with 1 additions and 1 deletions

View File

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