Explorar el Código

debug RTCDet-v2

yjh0410 hace 2 años
padre
commit
58e61efee3
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  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