update

config/model_config/rtcdet_config.py

@@ -40,7 +40,7 @@ rtcdet_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## Input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
+        'multi_scale': [0.7, 1.25], # 448 -> 800
         'trans_type': 'yolox_small',
         # ---------------- Assignment config ----------------
         ## Matcher
@@ -55,7 +55,7 @@ rtcdet_cfg = {
         'loss_cls_weight': 1.0,
         'loss_box_weight': 2.0,
         # ---------------- Train config ----------------
-        'trainer_type': 'rtmdet',
+        'trainer_type': 'yolox',
     },
 
     'rtcdet_l':{
@@ -96,7 +96,7 @@ rtcdet_cfg = {
         'head_depthwise': False,
         # ---------------- Train config ----------------
         ## Input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
+        'multi_scale': [0.7, 1.25], # 448 -> 800
         'trans_type': 'yolox_large',
         # ---------------- Assignment config ----------------
         ## Matcher
@@ -111,7 +111,7 @@ rtcdet_cfg = {
         'loss_cls_weight': 1.0,
         'loss_box_weight': 2.0,
         # ---------------- Train config ----------------
-        'trainer_type': 'rtmdet',
+        'trainer_type': 'yolox',
     },
 
 }

engine.py

@@ -389,7 +389,7 @@ class YoloxTrainer(object):
 
         # ---------------------------- Hyperparameters refer to YOLOX ----------------------------
         self.optimizer_dict = {'optimizer': 'sgd', 'momentum': 0.9, 'weight_decay': 5e-4, 'lr0': 0.01}
-        self.ema_dict = {'ema_decay': 0.9998, 'ema_tau': 2000}
+        self.ema_dict = {'ema_decay': 0.9999, 'ema_tau': 2000}
         self.lr_schedule_dict = {'scheduler': 'cosine', 'lrf': 0.05}
         self.warmup_dict = {'warmup_momentum': 0.8, 'warmup_bias_lr': 0.1}