소스 검색

update RTCDet

yjh0410 1 년 전
부모
커밋
06dd238cd2
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      train.sh

+ 1 - 1
train.sh

@@ -19,7 +19,7 @@ if [[ $MODEL == *"rtdetr"* ]]; then
     FIND_UNUSED_PARAMS=True
 elif [[ $MODEL == *"rtcdet"* ]]; then
     # Epoch setting
-    MAX_EPOCH=300
+    MAX_EPOCH=400
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20