yjh0410 il y a 1 an
Parent
commit
f49e8b9034
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      train.sh

+ 1 - 1
train.sh

@@ -7,7 +7,7 @@ DATA_ROOT="/data/datasets/"
 MODEL="yolox_s"
 IMAGE_SIZE=640
 # RESUME="weights/coco/yolov5_s/yolov5_s_best.pth"
-RESUME="none"
+RESUME="None"
 if [[ $MODEL == *"yolov8"* ]]; then
     # Epoch setting
     BATCH_SIZE=128