From a5bd0fe567db07a0f3a248a40c0935912233ed3b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 21 Apr 2020 12:19:29 -0700
Subject: [PATCH] tensorboard comment=opt.name

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index c4bab231..c002be25 100644
--- 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)