最近在用PyTorch Lightning重构之前的项目，真的太香了！

之前用原生PyTorch写训练代码，各种boilerplate code写得我头疼，什么GPU设置、梯度累积、学习率调度器...每次都要重复写一遍。

用了Lightning之后，只需要定义LightningModule，把forward、training_step、configure_optimizers这几个方法实现就行了，其他的框架都帮你搞定。

而且自带了很多实用功能：
- 自动混合精度训练
- 多GPU分布式训练
- TensorBoard集成
- 模型检查点保存

代码量至少减少了50%，而且更容易维护。强烈推荐给还在用原生PyTorch的朋友们！

GitHub: https://github.com/Lightning-AI/lightning