tensorboard comment=opt.name

2020-04-21 12:19:29 -07:00 · 2020-04-21 12:19:29 -07:00 · a5bd0fe567
parent 4c4f4f4dd4
commit a5bd0fe567
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -420,7 +420,7 @@ if __name__ == '__main__':
    tb_writer = None
    if not opt.evolve:  # Train normally
        print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
-        tb_writer = SummaryWriter()
+        tb_writer = SummaryWriter(comment=opt.name)
        train()  # train normally

    else:  # Evolve hyperparameters (optional)