Abstract

This is sample codes for hydra + mlflow + optuna

Slides & Articles

First slide for hydra + mlflow + optuna (ja)

Optuna Meetup #1 (ja)

Medium (en)

Medium (ja)

Requirements

Python 3.8.0

git clone https://github.com/supikiti/hydra-mlflow-optuna-sample.git
cd hydra-mlflow-optuna-sample
pip install -r requirements.txt

Sample run command

hydra + mlflow

python train.py -m model.node1=128,256 model.node2=256,512

hydra + mlflow + optuna

python3 train.py -m 'optimizer.lr=choice(0.1, 0.01, 0.001, 0.0001)' 'model.node1=range(10, 500)'

References

• https://ymym3412.hatenablog.com/entry/2020/02/09/034644
• https://cyberagent.ai/blog/research/12898/