🔊 [Add] supporting of tensorboard

yolo/config/general.yaml

@@ -7,7 +7,7 @@ out_path: runs
 exist_ok: True
 
 lucky_number: 10
-use_wandb: False
+use_wandb: True
 use_tensorboard: False
 
 weight: True # Path to weight or True for auto, False for no pretrained weight

yolo/utils/logging_utils.py

@@ -256,8 +256,10 @@ def setup(cfg: Config):
     progress.append(YOLORichProgressBar())
     progress.append(YOLORichModelSummary())
     progress.append(ImageLogger())
-
-    loggers.append(WandbLogger(project="YOLO", name=cfg.name, save_dir=save_path, id=None))
+    if cfg.use_tensorboard:
+        loggers.append(TensorBoardLogger(log_graph="all", save_dir=save_path))
+    if cfg.use_wandb:
+        loggers.append(WandbLogger(project="YOLO", name=cfg.name, save_dir=save_path, id=None))
 
     return progress, loggers