如何在W&B Sweep中基于扫参动态构造自定义命令行参数？

直接修改Sweep配置中的command部分，通过${parameters.参数名}的方式引用扫参的具体值，就能动态生成带参数的output_dir路径。

修改后的完整配置如下：

program: train.py
method: grid
parameters:
  batch_size:
    values: [8, 10, 12]
  lr:
    values: [0.0001, 0.001]
command:
  - ${env}
  - python3
  - ${program}
  - --output_dir /home/users/me/outputs/bs=${parameters.batch_size}_lr=${parameters.lr}
  - ${args}

关键说明： #

• ${parameters.batch_size}和${parameters.lr}会被W&B自动替换为当前sweep任务对应的参数值，比如当batch_size取8、lr取0.0001时，生成的命令行参数就是--output_dir /home/users/me/outputs/bs=8_lr=0.0001
• 这条自定义参数可以放在${args}的前面或后面，只要你的train.py脚本能正确解析--output_dir参数即可
• 注意参数名要和parameters字段下的键严格对应，比如这里用batch_size而不是bs，因为配置里定义的参数键是batch_size

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