fast-reid/scripts/debug.sh

7 lines
190 B
Bash

gpu=2
CUDA_VISIBLE_DEVICES=$gpu python tools/train_net.py -cfg='configs/softmax_triplet.yml' \
DATASETS.NAMES '("market1501",)' \
DATASETS.TEST_NAMES 'market1501' \
OUTPUT_DIR 'logs/test'