Hydra+MLflowで結果を保存する場所を指定する

概要

Hydra+MLflowで機械学習の結果をUIで確認するとき、どこに保存されるか分からなくなってしまった。
なので、結果を保存する場所を指定したい。

開発環境

python v3.6
MLflow v1.11.0
hydra-core v1.0.3

方法

こちらの方法に書いてある通り、Hydraの設定ファイル（yamlファイル）の最後に

hydra:
    job:
        env_set:
            MLFLOW_TRACKING_DIR: file://${hydra:runtime.cwd}/mlruns
            MLFLOW_TRACKING_URI: file://${hydra:runtime.cwd}/mlruns

を付け足したらいい。
保存する場所を変更するときには

export SAVE_PATH=file://$(pwd)/mlruns_ver2

# train.pyはHydraで書かれたPythonファイル
python train.py -m \  
                hydra.job.env_set.MLFLOW_TRACKING_DIR=${SAVE_PATH} \  
                hydra.job.env_set.MLFLOW_TRACKING_URI=${SAVE_PATH}