mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00

History

liaoxingyu 85cebe311f support hpo project

Summary: support BOHB search algorithm for reid.

2020-10-01 18:10:02 +08:00

support hpo project

2020-10-01 18:10:02 +08:00

support hpo project

2020-10-01 18:10:02 +08:00

README.md

support hpo project

2020-10-01 18:10:02 +08:00

train_hpo.py

support hpo project

2020-10-01 18:10:02 +08:00

Hyper-Parameter Optimization in FastReID

This project includes training reid models with hyper-parameter optimization.

Install the following

pip install 'ray[tune]'
pip install hpbandster ConfigSpace

Training

To train a model with BOHB, run

python3 projects/HPOReID/train_hpo.py --config-file projects/HPOReID/configs/baseline.yml

todo