fast-reid/scripts/train_softmax.sh
2018-08-02 23:18:09 +08:00

4 lines
174 B
Bash

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