Browse Source

train YOLOX-S with AdamW

yjh0410 1 year ago
parent
commit
8c075f5133
1 changed files with 1 additions and 0 deletions
  1. 1 0
      train.sh

+ 1 - 0
train.sh

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