python export.py \
  --weights /data6/zhangw/ZW_gra_project/yolov5_result_pretrained/G0_cub100_yolov5s6_raw_e300/weights/best.pt \
  --img 640 640 \
  --batch 1 \
  --device cpu \
  --include onnx \
  --simplify