Explorar o código

debug rtpdetr_r50

yjh0410 hai 1 ano
pai
achega
fe362e25c8
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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',
     },
 
 }