在修饰方法中访问 hydra 配置

Question

我想从修饰函数访问 hydra 配置，例如 sweeper：

@hydra.main(config_name="config")
def my_app(cfg: Config) -> None:
    OmegaConf.to_yaml(cfg.hydra.sweeper)

是否有任何参数可以传递给 @hydra.main() 以不删除该配置，或者是否有其他地方可以找到它？谢谢

Answer 1

from hydra.core.hydra_config import HydraConfig

@hydra.main()
def my_app(cfg: DictConfig) -> None:
    print(HydraConfig.get().sweeper)

你也可以通过插值配置访问：

config.yaml:

hydra_sweeper: ${hydra:sweeper}

参见this。