Explorar el Código

update RTCDet

yjh0410 hace 1 año
padre
commit
d16d4a5ae7
Se han modificado 1 ficheros con 6 adiciones y 0 borrados
  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