π [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)
|