write model description to tensorboard

2020-03-17 13:23:25 +01:00 · 2020-03-17 13:23:25 +01:00 · 141797b6ae
parent 2a15e39166
commit 141797b6ae
1 changed files with 2 additions and 3 deletions
--- a/train.py
+++ b/train.py
@ -692,7 +692,6 @@ 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)