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

add decoupled head

Browse Source
This commit is contained in:
Zhang Wei committed 2026-03-14 10:34:02 +08:00
1 parent 9b008eaa99
commit 14aeebc1e5
3 files changed
+134 -21
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

train_my_gra.sh
+4 -3
View File
@@ -1,9 +1,10 @@
export CUDA_VISIBLE_DEVICES=4,5,6,7
export CUDA_VISIBLE_DEVICES=0,1,2,3
python /data6/zhangw/yolov5/train.py \
--img 960 \
--batch 16 \
--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 G0_cub100_yolov5s6_raw_e300
--project /data6/zhangw/ZW_gra_project/yolov5_result_pretrained \
--name G6_cub100_yolov5s6_decoupled_head_e300 \
--cfg /data6/zhangw/yolov5/models/hub/yolov5s6_decoupled.yaml