From 141797b6ae44aa714adc70974c40204e0d4fa861 Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 17 Mar 2020 13:23:25 +0100 Subject: [PATCH] write model description to tensorboard --- train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index ea6d391c..62800c7b 100644 --- a/train.py +++ b/train.py @@ -692,12 +692,11 @@ if __name__ == '__main__': os.chmod(AUDIO_PATH, 0o775) os.chmod(OUT_PATH, 0o775) - if args.rank == 0: LOG_DIR = OUT_PATH tb_logger = Logger(LOG_DIR) - # write model desc to tensorboard - tb_logger.tb_add_text('model-description', c['run_description'], 0) + # write model desc to tensorboard + tb_logger.tb_add_text('model-description', c['run_description'], 0) try: main(args)