Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

change epochs to 300

Browse Source
This commit is contained in:
Zhang Wei committed 2025-12-22 15:42:01 +08:00
1 parent bc7d124645
commit 6ed5bc7dfe
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
train_my_project.sh
+2 -2
View File
@@ -2,9 +2,9 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3
python /data6/zhangw/yolov5/train.py \ python /data6/zhangw/yolov5/train.py \
--img 960 \ --img 960 \
--batch 16 \ --batch 16 \
--epochs 60 \ --epochs 300 \
--data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \ --data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
--weights /data6/zhangw/yolov5/yolov5s6.pt \ --weights /data6/zhangw/yolov5/yolov5s6.pt \
--project /data6/zhangw/ZW_gra_project/yolov5_result \ --project /data6/zhangw/ZW_gra_project/yolov5_result \
--name G1_cub100_yolov5s6_changeBackbone_e60 \ --name G1_cub100_yolov5s6_changeBackbone_e300 \
--cfg /data6/zhangw/yolov5/models/yolov5s.yaml --cfg /data6/zhangw/yolov5/models/yolov5s.yaml