Skip to content

Commit

Permalink
Check TensorBoard logger before adding graph (#8664)
Browse files Browse the repository at this point in the history
Otherwise, an error is thrown if the tensorboard logger is not included.
  • Loading branch information
JarnoRFB committed Jul 21, 2022
1 parent 92e47b8 commit 602d7ff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/loggers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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), [])
Expand Down

0 comments on commit 602d7ff

Please sign in to comment.