henry000 commited on
Commit
7e9db93
Β·
1 Parent(s): f4a5047

πŸ“ [Update] wandb logger name

Browse files
Files changed (1) hide show
  1. yolo/utils/logging_utils.py +3 -3
yolo/utils/logging_utils.py CHANGED
@@ -129,10 +129,10 @@ class ProgressLogger(Progress):
129
 
130
  def finish_one_epoch(self, batch_info: Dict[str, Any] = None, epoch_idx: int = -1):
131
  if self.task == "Train":
132
- prefix = "Loss/"
133
  elif self.task == "Validate":
134
- prefix = "Metrics/"
135
- batch_info = {f"{prefix}{key}": value for key, value in batch_info.items()}
136
  if self.use_wandb:
137
  self.wandb.log(batch_info, step=epoch_idx)
138
  self.remove_task(self.batch_task)
 
129
 
130
  def finish_one_epoch(self, batch_info: Dict[str, Any] = None, epoch_idx: int = -1):
131
  if self.task == "Train":
132
+ prefix = "Loss"
133
  elif self.task == "Validate":
134
+ prefix = "Metrics"
135
+ batch_info = {f"{prefix}/{key}": value for key, value in batch_info.items()}
136
  if self.use_wandb:
137
  self.wandb.log(batch_info, step=epoch_idx)
138
  self.remove_task(self.batch_task)