mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update training command line

pull/150/head
liaoxingyu 2020-06-16 19:45:58 +08:00
parent 8879db3fba
commit 2123cddb96
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GETTING_STARTED.md
View File

@ -21,7 +21,7 @@ You may want to use it as a reference to write your own training script.
To train a model with "train_net.py", first setup up the corresponding datasets following [datasets/README.md](https://github.com/JDAI-CV/fast-reid/tree/master/datasets), then run:
```bash
CUDA_VISIBLE_DEVICES=$gpus tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml
CUDA_VISIBLE_DEVICES=$gpus ./tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml
```
The configs are made for 1-GPU training.
@ -29,7 +29,7 @@ The configs are made for 1-GPU training.
To evaluate a model's performance, use
```bash
CUDA_VISIBLE_DEVICES=$gpus tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml \
CUDA_VISIBLE_DEVICES=$gpus ./tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml \
--eval-only MODEL.WEIGHTS /path/to/checkpoint_file
```