LoginSignup
1
1

More than 1 year has passed since last update.

Hydraのconfig nameを動的に設定・取得

Posted at

想定するファイル構成

├── conf
│   ├── transformer.json
│   └── lightgbm.json
│
├── main.py

config nameの動的な設定

python main.py --config-name=transformer train.cuda=0 ...

config nameの取得

from omegaconf import DictConfig
from hydra.core.hydra_config import HydraConfig


@hydra.main(version_base=None, config_path="conf", config_name=None)
def main(cfg: DictConfig):
    CONF_NAME = HydraConfig.get().job.config_name

1
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
1