소스 검색

debug rtpdetr_r50

yjh0410 1 년 전
부모
커밋
fe362e25c8
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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',
     },
 
 }