mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
25 lines
903 B
Bash
25 lines
903 B
Bash
#!/usr/bin/env 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.WITH_IBN 'True' \
|
|
MODEL.PRETRAIN_PATH '/home/user01/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar' \
|
|
MODEL.STAGE_WITH_GCB '(False, False, False, False)' \
|
|
SOLVER.LOSSTYPE '("softmax_smooth", "triplet")' \
|
|
OUTPUT_DIR 'logs/2019.8.25/duke/ibn_smooth'
|
|
|
|
#CUDA_VISIBLE_DEVICES=$gpu python tools/train.py -cfg='configs/softmax_triplet.yml' \
|
|
#DATASETS.NAMES '("duke",)' \
|
|
#DATASETS.TEST_NAMES 'duke' \
|
|
#MODEL.BACKBONE 'resnet50' \
|
|
#MODEL.WITH_IBN 'True' \
|
|
#MODEL.PRETRAIN_PATH '/home/user01/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar' \
|
|
#MODEL.STAGE_WITH_GCB '(False, True, True, True)' \
|
|
#SOLVER.LOSSTYPE '("softmax_smooth", "triplet")' \
|
|
#OUTPUT_DIR 'logs/2019.8.25/duke/ibn_11_gcnet_smooth'
|
|
|
|
|