train_my_gra.sh

2025-12-24 13:51:08 +08:00
1 parent 7d7084b857
commit e10e5dd65a
@@ -6,4 +6,5 @@ python /data6/zhangw/yolov5/train.py \
    --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
+    --name G0_cub100_yolov5s6_raw_e300 \
+    --cfg /data6/zhangw/yolov5/models/yolov5s.yaml