From 6ed5bc7dfe58737dbfea435493af1ec3bd158537 Mon Sep 17 00:00:00 2001 From: Zhang Wei <3160665918@qq.com> Date: Mon, 22 Dec 2025 15:42:01 +0800 Subject: [PATCH] change epochs to 300 --- train_my_project.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_my_project.sh b/train_my_project.sh index f708ca2..6926a8e 100644 --- a/train_my_project.sh +++ b/train_my_project.sh @@ -2,9 +2,9 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3 python /data6/zhangw/yolov5/train.py \ --img 960 \ --batch 16 \ - --epochs 60 \ + --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 G1_cub100_yolov5s6_changeBackbone_e60 \ + --name G1_cub100_yolov5s6_changeBackbone_e300 \ --cfg /data6/zhangw/yolov5/models/yolov5s.yaml \ No newline at end of file