differentiate modules on TB with a prefix

This commit is contained in:
erogol 2020-05-30 18:07:01 +02:00
parent a2e74c58f9
commit 3034797211
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)