Hydra 配置框架

Meta 开源的 Python 配置管理框架，用 YAML 组合配置、命令行随时覆盖参数。

Hydra 是 Meta（原 Facebook AI Research）开源的 Python 配置框架，底层用 OmegaConf 读写 YAML。它把一次实验的配置拆成多个可组合的小文件（模型、数据集、任务各一份），运行时按需拼装，并允许在命令行直接覆盖任意字段，还能用 multirun 一次扫一组超参数（超参数：训练前人为设定的值，如学习率）。机器人学习代码里参数多、实验多，Hydra 让「改一个参数重跑」不必改代码，并自动为每次运行建输出目录、保存当时的完整配置，便于复现。Diffusion Policy、Isaac Lab 等代码库都用它管理训练配置。

例子在 Diffusion Policy 仓库里执行 python train.py --config-name=train_diffusion_unet_image_workspace task=pusht_image training.seed=42，就能在不改代码的情况下换任务、换随机种子。

也叫

facebookresearch/hydra

相关

超参数(Hyperparameter)、随机种子与可复现性(Random Seed & Reproducibility)、扩散策略(Diffusion Policy)、Isaac Lab、Weights & Biases

来源

Hydra 官网
facebookresearch/hydra (GitHub)