fast-reid/scripts/train_benchmark.sh

4 lines
103 B
Bash

GPUS=0,1,2,3
CUDA_VISIBLE_DEVICES=$GPUS python tools/train_net.py -cfg='configs/resnet_benchmark.yml'