diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
index 42b696b..88bdb05 100644
--- a/utils/loggers/__init__.py
+++ b/utils/loggers/__init__.py
@@ -102,7 +102,7 @@ class Loggers():
         # 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), [])