Spaces:

henry000
/

YOLO

Running

henry000 commited on Jun 5, 2024

Commit

ce50967

1 Parent(s): 97e9dcb

🚑️ [Fix] logging error, cfg->exp_name

Files changed (1) hide show

yolo/utils/logging_utils.py CHANGED Viewed

@@ -50,7 +50,7 @@ class ProgressTracker:
         if self.use_wandb:
             wandb.errors.term._log = custom_wandb_log
             self.wandb = wandb.init(
-                project="YOLO", resume="allow", mode="online", dir=save_path, id=None, name=cfg.name
             )
     def start_train(self, num_epochs: int):

         if self.use_wandb:
             wandb.errors.term._log = custom_wandb_log
             self.wandb = wandb.init(
+                project="YOLO", resume="allow", mode="online", dir=save_path, id=None, name=exp_name
             )
     def start_train(self, num_epochs: int):