Pytorch Lightning

一、项目简介

PyTorch Lightning 是 PyTorch 的训练组织框架。它把训练循环中的样板代码（设备搬运、分布式、混合精度、日志、checkpoint）抽象掉，让研究者专注模型与数据，同时保留 PyTorch 的灵活性。

二、核心特性

• 减少样板代码：不用手写训练循环；
• 训练流程标准化：统一的模块与回调；
• 多 GPU / 分布式开箱：改参数即可扩展；
• 内置日志与 checkpoint；
• 混合精度支持；
• 与 PyTorch 生态兼容。

三、适用场景

• 模型训练组织与快速实验；
• 分布式训练；
• 研究原型开发。

四、快速上手

pip install lightning
import lightning as L
class MyModel(L.LightningModule): ...
trainer = L.Trainer(max_epochs=5, accelerator="gpu")
trainer.fit(model, dataloader)

五、注意事项

• 抽象层需时间适应；
• 复杂自定义逻辑需理解框架机制；
• 注意版本匹配。