YOLOv3-model-pruning/xs_train.sh

4 lines
109 B
Bash

#!bash
CARD=0,1
CUDA_VISIBLE_DEVICES=$CARD python train.py --model_def config/yolov3-person.cfg -sr --s 0.01