fast-reid/scripts/train_mask.sh

4 lines
97 B
Bash

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