fast-reid/scripts/train_duke.sh

9 lines
213 B
Bash
Raw Normal View History

2019-08-20 09:36:47 +08:00
gpu=0
CUDA_VISIBLE_DEVICES=$gpu python tools/train.py -cfg='configs/softmax.yml' \
2019-08-20 09:36:47 +08:00
DATASETS.NAMES '("duke",)' \
DATASETS.TEST_NAMES 'duke' \
MODEL.BACKBONE 'resnet50' \
2019-08-23 07:49:03 +08:00
MODEL.IBN 'True' \
OUTPUT_DIR 'logs/test'