change epochs to 300
This commit is contained in:
1 parent
bc7d124645
commit
6ed5bc7dfe
1 file changed
+2
-2
+2
-2
@@ -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
|
||||
Reference in new issue
Block a user