mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
21 lines
389 B
Markdown
21 lines
389 B
Markdown
|
# Hyper-Parameter Optimization in FastReID
|
||
|
|
|
||
|
|
This project includes training reid models with hyper-parameter optimization.
|
||
|
|
|
||
|
|
Install the following
|
||
|
|
|
||
|
|
```bash
|
||
|
|
pip install 'ray[tune]'
|
||
|
|
pip install hpbandster ConfigSpace
|
||
|
|
```
|
||
|
|
|
||
|
|
## Training
|
||
|
|
|
||
|
|
To train a model with `BOHB`, run
|
||
|
|
|
||
|
|
```bash
|
||
|
|
python3 projects/HPOReID/train_hpo.py --config-file projects/HPOReID/configs/baseline.yml
|
||
|
|
```
|
||
|
|
|
||
|
|
## Known issues
|
||
|
|
todo
|