fast-reid/scripts/train_triplet_softmax.sh

5 lines
191 B
Bash

#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES=0 python3 main_reid.py train --save_dir='./pytorch-ckpt/market1501_softmax_triplet' \
--max_epoch=400 --eval_step=5 --model_name='softmax_triplet' \