update

train.sh

@@ -13,42 +13,49 @@ if [[ $MODEL == *"yolov8"* ]]; then
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"rtcdet"* ]]; then
+elif [[ $MODEL == *"yolox"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"yolox"* ]]; then
+elif [[ $MODEL == *"yolov7"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"yolov7"* ]]; then
+elif [[ $MODEL == *"yolov5"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"yolov5"* ]]; then
+elif [[ $MODEL == *"yolov4"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"yolov4"* ]]; then
+elif [[ $MODEL == *"yolov3"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300
     WP_EPOCH=3
     EVAL_EPOCH=10
     NO_AUG_EPOCH=20
-elif [[ $MODEL == *"yolov3"* ]]; then
+elif [[ $MODEL == *"rtcdet"* ]]; then
+    # Epoch setting
+    BATCH_SIZE=128
+    MAX_EPOCH=300
+    WP_EPOCH=3
+    EVAL_EPOCH=10
+    NO_AUG_EPOCH=20
+elif [[ $MODEL == *"ctrnet"* ]]; then
     # Epoch setting
     BATCH_SIZE=128
     MAX_EPOCH=300