export CUDA_VISIBLE_DEVICES=4,5,6,7
python /data6/zhangw/yolov5/train.py \
    --img 960 \
    --batch 16 \
    --epochs 300 \
    --data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
    --weights /data6/zhangw/yolov5/yolov5s6.pt \
    --project /data6/zhangw/ZW_gra_project/yolov5_result_pretrained \
    --name G10_decoupled_head_light_FGEM_safe_e300 \
    --cfg /data6/zhangw/yolov5/models/yolov5s6_decoupled_light_FGEM_safe.yaml