在修饰的方法中访问hydra配置

Question

我想从装饰过的函数中访问九头蛇配置，例如，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。