README.md 572 Bytes
Newer Older
dengjb's avatar
update  
dengjb committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# 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 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

```bash
python3 projects/FastTune/tune_net.py --config-file projects/FastTune/configs/search_trial.yml --srch-algo "bohb"
```

## Known issues
todo