Bladeren bron

update RTCDet

yjh0410 1 jaar geleden
bovenliggende
commit
d16d4a5ae7
1 gewijzigde bestanden met toevoegingen van 6 en 0 verwijderingen
  1. 6 0
      train.sh

+ 6 - 0
train.sh

@@ -17,6 +17,12 @@ if [[ $MODEL == *"rtdetr"* ]]; then
     EVAL_EPOCH=1
     NO_AUG_EPOCH=-1
     FIND_UNUSED_PARAMS=True
+elif [[ $MODEL == *"rtcdet"* ]]; then
+    # Epoch setting
+    MAX_EPOCH=300
+    WP_EPOCH=3
+    EVAL_EPOCH=10
+    NO_AUG_EPOCH=20
 elif [[ $MODEL == *"yolov8"* ]]; then
     # Epoch setting
     MAX_EPOCH=500