diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 42b696ba644f..88bdb0521619 100644 --- 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), [])