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