YOLOv3-model-pruning/train_person.sh

4 lines
96 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