YOLOv3-model-pruning/xs_train.sh

4 lines
109 B
Bash
Raw Normal View History

2024-06-25 14:07:50 +08:00
#!bash
CARD=0,1
CUDA_VISIBLE_DEVICES=$CARD python train.py --model_def config/yolov3-person.cfg -sr --s 0.01