Explorar o código

use qfl cls loss

yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
b4c63ade7d
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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,