Explorar el Código

debug rtpdetr_r50

yjh0410 hace 1 año
padre
commit
fe362e25c8
Se han modificado 1 ficheros con 1 adiciones y 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',
     },
 
 }