深度学习助手

PyTorch Lightning 是一个深度学习框架，旨在简化 PyTorch 代码的组织。通过将代码结构化为 LightningModules，支持多 GPU/TPU 训练，提供数据管道、回调和日志记录功能，适用于可扩展的神经网络训练。