Hydra instantiate目标时如何将参数设置为2的幂次？

Hydra底层基于OmegaConf实现配置管理，YAML本身不支持直接解析**幂运算表达式，直接填写会被识别为字符串，你可以通过以下两种方式解决： #

方法1：注册自定义幂运算解析器（安全推荐） #

先在代码入口处注册自定义的OmegaConf解析器，用于计算幂运算结果：

import hydra
import hydra.utils as hu
from omegaconf import OmegaConf

# 注册幂运算解析器
OmegaConf.register_new_resolver("pow", lambda base, exponent: base ** exponent)

@hydra.main(config_path='conf', config_name='config')
def main(cfg):
    vec = hu.instantiate(cfg.vec)
    vec.transform(['Erroneous Monk'])

if __name__ == "__main__":
    main()

修改config.yaml，用插值语法调用解析器：

vec:
  _target_: sklearn.feature_extraction.text.HashingVectorizer
  n_features: ${pow:2,18}

运行时会自动计算得到整数262144传入参数，不会触发类型错误。 #

方法2：使用内置eval解析器（仅适合本地调试） #

如果不想修改代码，也可以直接用OmegaConf自带的eval解析器执行表达式，仅修改配置文件即可：

vec:
  _target_: sklearn.feature_extraction.text.HashingVectorizer
  n_features: ${eval:'2**18'}

注意该方法存在任意代码执行的安全风险，不要在生产环境或者配置来源不可信的场景使用。 #

内容的提问来源于stack exchange，提问作者Grr