从python文件在hydra DictConfig中创建一个新键

Question

我想在我的Hydra配置加载后添加一个键+值。本质上，我想运行我的代码并检查gpu是否可用。如果是，将设备日志为gpu，否则保留cpu。

基本上节省了:

的输出

torch.cuda.is_available()

当我试图添加一个键"setdefault"比如:

@hydra.main(config_path="conf", config_name="config")
def my_app(cfg: DictConfig) -> None:
cfg.setdefault("new_key", "new_value")

同样的错误如果我做:

cfg.new_key = "new_value"
print(cfg.new_key)

我得到错误:

omegaconf.errors.ConfigKeyError: Key 'new_key' is not in struct
full_key: new_key
reference_type=Optional[Dict[Union[str, Enum], Any]]
object_type=dict

我目前的解决办法是只使用OmegaConfig这样:

cfg = OmegaConf.structured(OmegaConf.to_yaml(cfg))
cfg.new_key = "new_value"
print(cfg.new_key)
>>>> new_value

肯定有更好的方法来做这件事吧?

Answer 1

Hydra在它生成的OmegaConf配置对象的根上设置结构标志。有关struct标志的更多信息，请参阅此处。

您可以使用open_dict()暂时禁用此功能，并允许添加新键:

>>> from omegaconf import OmegaConf,open_dict
>>> conf = OmegaConf.create({"a": {"aa": 10, "bb": 20}})
>>> OmegaConf.set_struct(conf, True)
>>> with open_dict(conf):
...   conf.a.cc = 30
>>> conf.a.cc
30