Bläddra i källkod

debug RTCDet-v2

yjh0410 2 år sedan
förälder
incheckning
58e61efee3
1 ändrade filer med 2 tillägg och 2 borttagningar
  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