changed names

luxonis · May 21, 2024 · 35b17d7 · 35b17d7
1 parent 6cfe1ea
commit 35b17d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/luxonis_train/callbacks/gpu_stats_monitor.py b/luxonis_train/callbacks/gpu_stats_monitor.py
@@ -253,7 +253,7 @@ def _parse_gpu_stats(
             for j, (x, unit) in enumerate(keys):
                 if unit == "%":
                     unit = "percent"
-                logs[f"GPU {device_id}/{x} {unit}"] = stats[i][j]
+                logs[f"GPU_{device_id}/{x} - {unit}"] = stats[i][j]
         return logs
 
     def _get_gpu_stat_keys(self) -> List[Tuple[str, str]]: