Browse Source

use 6 decoder layers for R18

yjh0410 1 year ago
parent
commit
dc43a8b025
1 changed files with 1 additions and 1 deletions
  1. 1 1
      config/model_config/rtdetr_config.py

+ 1 - 1
config/model_config/rtdetr_config.py

@@ -52,7 +52,7 @@ rtdetr_cfg = {
                        'giou': 2,},
         # ---------------- Train config ----------------
         ## input
-        'multi_scale': [0.5, 1.5],   # 320 -> 960
+        'multi_scale': [0.5, 1.25],   # 320 -> 800
         'trans_type': 'rtdetr_s',
         # ---------------- Train config ----------------
         'trainer_type': 'rtdetr',