yjh0410 2 vuotta sitten
vanhempi
sitoutus
4cedda130c
1 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  1. 2 2
      config/model_config/rtcdet_v2_config.py

+ 2 - 2
config/model_config/rtcdet_v2_config.py

@@ -53,12 +53,12 @@ rtcdet_v2_cfg = {
         # ---------------- Loss config ----------------
         ## Loss weight
         'ema_update': False,
-        'loss_box_aux': False,
+        'loss_box_aux': True,
         'loss_cls_weight': {'tal': 0.5, 'ota': 1.0},
         'loss_box_weight': {'tal': 7.0, 'ota': 5.0},
         'loss_dfl_weight': {'tal': 1.5, 'ota': 1.0},
         # ---------------- Train config ----------------
-        'trainer_type': 'rtmdet',
+        'trainer_type': 'yolox',
     },
 
 }