9 lines
312 B
Bash
9 lines
312 B
Bash
export CUDA_VISIBLE_DEVICES=5,6
|
|
python /data6/zhangw/yolov5_birdproject/train.py \
|
|
--img 960 \
|
|
--batch 16 \
|
|
--epochs 100 \
|
|
--data /data6/zhangw/yolov5_birdproject/data/coco128.yaml \
|
|
--weights /data6/zhangw/yolov5_birdproject/yolov5s6.pt \
|
|
--name KTY_yolov5s6_fromYNDC_addKTYneg_e100 \
|
|
--device 5,6 |