<neck>change concat to weightedadd

This commit is contained in:
Zhang Wei committed 2025-12-25 09:25:37 +08:00
1 parent 434a7ade3f
commit 492ad7a7fe
2 files changed
+27 -23

No files matched your search

+3 -4
View File
@@ -2,9 +2,8 @@ export CUDA_VISIBLE_DEVICES=4,5,6,7
python /data6/zhangw/yolov5/train.py \
--img 960 \
--batch 16 \
--epochs 300 \
--epochs 500 \
--data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
--weights /data6/zhangw/yolov5/yolov5s6.pt \
--project /data6/zhangw/ZW_gra_project/yolov5_result \
--project /data6/zhangw/ZW_gra_project/yolov5_result_structure \
--cfg /data6/zhangw/yolov5/models/yolov5s.yaml \
--name G2_2_cub100_yolov5s6_changeNeck_e300
--name G2_2_cub100_yolov5s6_changeNeck_e500