yjh0410 преди 2 години
родител
ревизия
b4c63ade7d
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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,