π [Update] wandb logger name
Browse files
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)
|