[fix] change yolox_loss to yolov9_loss

Nota-NetsPresso · Dec 6, 2024 · 62572b8 · 62572b8
1 parent 3bb5f5d
commit 62572b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/config/model/yolo/yolov9_c-detection.yaml b/config/model/yolo/yolov9_c-detection.yaml
@@ -66,6 +66,6 @@ model:
       nms_thresh: 0.65
       class_agnostic: false
   losses:
-    - criterion: yolox_loss
+    - criterion: yolov9_loss
       weight: ~
-      l1_activate_epoch: 285
+      l1_activate_epoch: ~
diff --git a/config/model/yolo/yolov9_m-detection.yaml b/config/model/yolo/yolov9_m-detection.yaml
@@ -66,6 +66,6 @@ model:
       nms_thresh: 0.65
       class_agnostic: false
   losses:
-    - criterion: yolox_loss
+    - criterion: yolov9_loss
       weight: ~
-      l1_activate_epoch: 285
+      l1_activate_epoch: ~