modify load_cache func

train.py

@@ -89,7 +89,7 @@ def parse_args():
                         help='data root')
     parser.add_argument('-d', '--dataset', default='coco',
                         help='coco, voc, widerface, crowdhuman')
-    parser.add_argument('--load_cache', type=str, default=None,
+    parser.add_argument('--load_cache', action='store_true', default=False,
                         help='Path to the cached data.')
     parser.add_argument('--num_workers', default=4, type=int, 
                         help='Number of workers used in dataloading')

train.sh

@@ -2,7 +2,6 @@
 DATASET="coco"
 DATA_ROOT="/data/datasets/"
 # DATA_ROOT="/Users/liuhaoran/Desktop/python_work/object-detection/dataset/"
-CACHED_DATA="${DATA_ROOT}/coco_train.pth"
 
 # MODEL setting
 MODEL="yolov8_l"
@@ -59,7 +58,6 @@ if [ $WORLD_SIZE == 1 ]; then
             --cuda \
             --dataset ${DATASET} \
             --root ${DATA_ROOT} \
-            --load_cache ${CACHED_DATA} \
             --model ${MODEL} \
             --batch_size ${BATCH_SIZE} \
             --img_size ${IMAGE_SIZE} \
@@ -77,7 +75,6 @@ elif [[ $WORLD_SIZE -gt 1 && $WORLD_SIZE -le 8 ]]; then
             -dist \
             --dataset ${DATASET} \
             --root ${DATA_ROOT} \
-            --load_cache ${CACHED_DATA} \
             --model ${MODEL} \
             --batch_size ${BATCH_SIZE} \
             --img_size ${IMAGE_SIZE} \