如何从组配置而非全局配置指定Hydra Multirun参数扫描？

问题背景 #

现有目录结构如下：

├── conf
│   ├── config.yaml
│   └── test
│       ├── imagenet.yaml
│       └── mnist.yaml
└── main.py

需求是用Hydra Multirun功能，为不同test组配置设置专属的lambda参数扫描：

• imagenet测试需运行lambda=0和lambda=100
• MNIST测试需运行lambda=0和lambda=1
  执行python main.py test=glob(*)后期望生成4个任务，但当前组配置内的sweeper参数未被Hydra扫描器识别。

核心原因 #

Hydra的Multirun扫描器仅读取全局配置或命令行传入的sweeper规则，组配置文件内的hydra块属于普通配置内容，不会被扫描器解析执行。

两种可行解决方案 #

方案一：命令行直接指定组合规则 #

直接在命令行中明确每个组配置对应的参数范围，用引号包裹每组的配置+参数组合，最后加上-m启用Multirun：

python main.py 'test=imagenet,lambda=0,100' 'test=mnist,lambda=0,1' -m

该方式直观灵活，无需修改现有组配置文件，适合快速测试。

方案二：全局配置结合条件映射 #

通过全局配置定义不同test对应的lambda扫描范围，让Hydra根据选择的test自动加载对应参数：

1. 修改conf/config.yaml：

defaults:
  - test: ???
  - _self_

hydra:
  mode: MULTIRUN
  sweeper:
    params:
      lambda: ${test.lambda_values}

# 映射每个test对应的lambda扫描范围
test:
  imagenet:
    lambda_values: 0,100
  mnist:
    lambda_values: 0,1

2. 清空组配置内的hydra块：
   修改conf/test/imagenet.yaml：

foo: bar
beep: boop

修改conf/test/mnist.yaml：

baz: baq
meep: moop

3. 执行原命令即可：

python main.py test=glob(*)

该方式更具可维护性，适合后续扩展更多测试组的参数规则。

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