yjh0410 1 rok pred
rodič
commit
d16d4a5ae7
1 zmenil súbory, kde vykonal 6 pridanie a 0 odobranie
  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