10 lines
425 B
Bash
10 lines
425 B
Bash
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_pretrained \
|
|
--name G6_cub100_yolov5s6_decoupled_head_e300 \
|
|
--cfg /data6/zhangw/yolov5/models/hub/yolov5s6_decoupled.yaml |