Kaynağa Gözat

debug rtpdetr_r50

yjh0410 1 yıl önce
ebeveyn
işleme
fe362e25c8
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      config/model_config/rtpdetr_config.py

+ 1 - 1
config/model_config/rtpdetr_config.py

@@ -51,7 +51,7 @@ rtpdetr_cfg = {
         'multi_scale': [0.5, 1.25],   # 320 -> 800
         'trans_type': 'rtdetr_l',
         # ---------------- Train config ----------------
-        'trainer_type': 'rtdetr',
+        'trainer_type': 'rtpdetr',
     },
 
 }