diff --git a/train_my_gra.sh b/train_my_gra.sh
index 774e767..f20bf38 100644
--- a/train_my_gra.sh
+++ b/train_my_gra.sh
@@ -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
\ No newline at end of file
+    --name G0_cub100_yolov5s6_raw_e300 \
+    --cfg /data6/zhangw/yolov5/models/yolov5s.yaml
\ No newline at end of file