From 5ff6e6b3a56d8c271e4ab6947b4af1b31f813942 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 9 Aug 2019 19:35:02 +0200
Subject: [PATCH] tensorboard updates

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

diff --git a/train.py b/train.py
index c9830b85..0d1aa45d 100644
--- a/train.py
+++ b/train.py
@@ -352,6 +352,7 @@ if __name__ == '__main__':
     opt = parser.parse_args()
     print(opt)
 
+    tb_writer = None
     if not opt.evolve:  # Train normally
         try:
             # Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/
@@ -359,7 +360,7 @@ if __name__ == '__main__':
 
             tb_writer = SummaryWriter()
         except:
-            tb_writer = None
+            pass
 
         results = train(opt.cfg,
                         opt.data,