History

liaoxingyu e5c3ff3b5b fix typro in hpo training		2020-10-10 18:55:32 +08:00
..
configs	fix typro in hpo training	2020-10-10 18:55:32 +08:00
hporeid	add more hpo search algorithms	2020-10-10 17:35:25 +08:00
README.md	add more hpo search algorithms	2020-10-10 17:35:25 +08:00
train_hpo.py	fix typro in hpo training	2020-10-10 18:55:32 +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 hyperopt

Example

This is an example for tuning batch_size and num_instance automatically.

To train hyperparameter optimization with BOHB(Bayesian Optimization with HyperBand) search algorithm, run

python3 projects/HPOReID/train_hpo.py --config-file projects/HPOReID/configs/baseline.yml --srch-algo "bohb"

todo