diff --git a/train_my_gra.sh b/train_my_gra.sh new file mode 100644 index 0000000..774e767 --- /dev/null +++ b/train_my_gra.sh @@ -0,0 +1,9 @@ +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 \ + --name G0_cub100_yolov5s6_raw_e300 \ No newline at end of file