use 6 decoder layers for R18

config/model_config/rtdetr_config.py

@@ -30,7 +30,7 @@ rtdetr_cfg = {
         # Transformer Decoder
         'transformer': 'rtdetr_transformer',
         'de_num_heads': 8,
-        'de_num_layers': 6,
+        'de_num_layers': 3,
         'de_ffn_dim': 1024,
         'de_dropout': 0.0,
         'de_act': 'relu',
@@ -52,7 +52,7 @@ rtdetr_cfg = {
                        'giou': 2,},
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
+        'multi_scale': [0.5, 1.5],   # 320 -> 960
         'trans_type': 'rtdetr_s',
         # ---------------- Train config ----------------
         'trainer_type': 'rtdetr',