fast-reid/scripts/train_duke.sh

25 lines
903 B
Bash
Raw Normal View History

2019-08-26 23:10:08 +08:00
#!/usr/bin/env bash
2019-08-20 09:36:47 +08:00
gpu=0
2019-08-26 23:10:08 +08:00
CUDA_VISIBLE_DEVICES=$gpu python tools/train.py -cfg='configs/softmax_triplet.yml' \
2019-08-20 09:36:47 +08:00
DATASETS.NAMES '("duke",)' \
DATASETS.TEST_NAMES 'duke' \
MODEL.BACKBONE 'resnet50' \
2019-08-26 23:10:08 +08:00
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'