yjh0410 2 rokov pred
rodič
commit
b4c63ade7d
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  1. 1 1
      config/model_config/rtcdet_config.py

+ 1 - 1
config/model_config/rtcdet_config.py

@@ -217,7 +217,7 @@ rtcdet_cfg = {
         'matcher_hpy': {'center_sampling_radius': 2.5,
                         'topk_candidate': 10},
         # ---------------- Loss config ----------------
-        'cls_loss': 'qfl',
+        'cls_loss': 'bce',
         'loss_cls_weight': 1.0,
         'loss_dfl_weight': 1.0,
         'loss_box_weight': 5.0,