通过Hydra 配置文件实例化Lightning类

通过 Hydra 的配置文件灵活控制 LightningModule、LightningDataModule、Callback 和 Trainer 的实例化过程。

Lightning 与 Hydra 的结合

1. 配置文件结构

一个典型的配置文件目录可能如下：

configs/
├── callbacks/
│   ├── early_stopping.yaml
│   ├── checkpoint.yaml
│   └── default.yaml
├── data/
│   ├── openfold.yaml
│   └── cifar10.yaml
├── model/
│   ├── alphafold3.yaml
│   └── resnet.yaml
├── trainer/
│   ├── deepspeed.yaml
│   └── default.yaml
├── logger/
│   ├── wandb.yaml
│   └── tensorboard.yaml
└── main.yaml

main.yaml:

defaults:- data: openfold- model: alphafold3- logger: wandb- callbacks: default- trainer: default

data/openfold.yaml:

_target_: my_project.data.OpenFoldDataModule
batch_size: 32
num_workers: 4