Explorar el Código

modify YOLOv8's config

yjh0410 hace 2 años
padre
commit
1735d40e7a
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      config/model_config/yolov8_config.py

+ 1 - 1
config/model_config/yolov8_config.py

@@ -99,7 +99,7 @@ yolov8_cfg = {
         'loss_box_weight': 7.5,
         'loss_dfl_weight': 1.5,
         # ---------------- Train config ----------------
-        'trainer_type': 'yolov8',
+        'trainer_type': 'rtcdet',
     },
 
 }