train YOLOv5 on COCO with 250 epoch

test.py

@@ -123,7 +123,7 @@ def test(args,
     save_path = os.path.join('det_results/', args.dataset, args.model)
     os.makedirs(save_path, exist_ok=True)
 
-    for index in range(num_images):
+    for index in range(1200, num_images):
         print('Testing image {:d}/{:d}....'.format(index+1, num_images))
         image, _ = dataset.pull_image(index)
 

train.sh

@@ -7,7 +7,7 @@ python train.py \
         -bs 16 \
         -size 640 \
         --wp_epoch 1 \
-        --max_epoch 300 \
+        --max_epoch 250 \
         --eval_epoch 10 \
         --ema \
         --fp16 \