Hydra:使用 Compose API 覆盖 hydra.run.dir/working 目录管理

问题描述

我正在尝试使用 Hydra 的 compose API 以编程方式而不是通过 CLI 启动运行。这在大多数情况下有效。但是,在使用 compose API 时,覆盖 hydra.run.dir 以更改基本目录似乎没有效果。即:

    with hydra.experimental.initialize_config_module(config_module=config_module):
        cfg = hydra.experimental.compose(
            config_name=config_name,overrides=["hydra.run.dir=/tmp/workdir",...],return_hydra_config=True
        )

        hydra.core.hydra_config.HydraConfig.instance().set_config(cfg)

        with omegaconf.open_dict(cfg):
            del cfg["hydra"]

使用 DictConfig 的相应条目生成 hydra.run.dir,但工作目录未更改。

compose API 文档指出,不使用 @hydra.main 会导致 Hydra 的工作目录管理失效。有解决方法吗?

解决方法

Compose API 是无状态的,故意不更改工作目录、配置日志记录或更改全局状态,也不与命令行集成。如果您需要这些功能,您应该考虑使用 Main-Class: Main Class-Path: lib/abc.jar

作为一种解决方法,您可以在您的终端以编程方式调用 chdir(使用 os.chdir)。您可能还需要先 mkdir。