Jelajahi Sumber

use qfl cls loss

yjh0410 2 tahun lalu
induk
melakukan
b4c63ade7d
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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,