diff --git a/config/model/yolo/yolov9_c-detection.yaml b/config/model/yolo/yolov9_c-detection.yaml index ae2bbe95..7bfcd5eb 100644 --- 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 index e0a85a78..cd9f85e7 100644 --- 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: ~