History

liaoxingyu 91ff631184 Minor changes Some minor changes, such as class name changing, remove extra blank line, etc.		2021-05-31 17:27:14 +08:00
..
autotuner	update fastreid V1.0	2021-01-18 11:36:38 +08:00
configs	Minor changes	2021-05-31 17:27:14 +08:00
README.md	update fastreid V1.0	2021-01-18 11:36:38 +08:00
tune_net.py	update fastreid V1.0	2021-01-18 11:36:38 +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/FastTune/tune_net.py --config-file projects/FastTune/configs/search_trial.yml --srch-algo "bohb"

todo