瀏覽代碼

debug RTCDet-v2

yjh0410 2 年之前
父節點
當前提交
58e61efee3
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      config/model_config/rtcdet_v2_config.py

+ 2 - 2
config/model_config/rtcdet_v2_config.py

@@ -50,7 +50,7 @@ rtcdet_v2_cfg = {
                             'beta': 6.0},
                     'ota': {'center_sampling_radius': 2.5,
                              'topk_candidate': 10},
-                    'switch_epoch': 1,
+                    'switch_epoch': 5,
                     },
         # ---------------- Loss config ----------------
         ## Loss weight
@@ -110,7 +110,7 @@ rtcdet_v2_cfg = {
                             'beta': 6.0},
                     'ota': {'center_sampling_radius': 2.5,
                              'topk_candidate': 10},
-                    'switch_epoch': 1,
+                    'switch_epoch': 5,
                     },
         # ---------------- Loss config ----------------
         ## Loss weight