fast-reid/scripts/train_market_bj.sh

5 lines
173 B
Bash

gpu=3
CUDA_VISIBLE_DEVICES=$gpu python tools/train.py -cfg='configs/softmax_triplet.yml' \
DATASETS.NAMES '("market1501","beijing",)' \
OUTPUT_DIR 'logs/beijing/market_bj'