mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
11 lines
287 B
Bash
11 lines
287 B
Bash
|
gpu=0
|
||
|
|
|
||
|
|
CUDA_VISIBLE_DEVICES=$gpu python tools/train.py -cfg='configs/softmax_triplet.yml' \
|
||
|
|
DATASETS.NAMES '("duke",)' \
|
||
|
|
DATASETS.TEST_NAMES 'duke' \
|
||
|
|
MODEL.BACKBONE 'resnet50' \
|
||
|
|
MODEL.IBN 'False' \
|
||
|
|
INPUT.DO_LIGHTING 'False' \
|
||
|
|
SOLVER.OPT 'adam' \
|
||
|
|
OUTPUT_DIR 'logs/2019.8.19/duke/resnet'
|