Check TensorBoard logger before adding graph (ultralytics#8664)

Otherwise, an error is thrown if the tensorboard logger is not included.
ctjanuhowski · Sep 8, 2022 · fc7a251 · fc7a251
1 parent b12c68c
commit fc7a251
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -102,7 +102,7 @@ def on_train_batch_end(self, ni, model, imgs, targets, paths, plots):
         # Callback runs on train batch end
         if plots:
             if ni == 0:
-                if not self.opt.sync_bn:  # --sync known issue https://github.com/ultralytics/yolov5/issues/3754
+                if self.tb and not self.opt.sync_bn:  # --sync known issue https://github.com/ultralytics/yolov5/issues/3754
                     with warnings.catch_warnings():
                         warnings.simplefilter('ignore')  # suppress jit trace warning
                         self.tb.add_graph(torch.jit.trace(de_parallel(model), imgs[0:1], strict=False), [])