闻韶

还记得那个看起来像 Keras 的轻量版 PyTorch 框架 Lightning 吗？它终于出了 1.0.0 版本，并增添了很多新功能，在度量、优化、日志记录、数据流、检查点等方面均进行了完善。Keras 和 PyTorch 都是对初学者非常友好的深度学习框架，两者各有优势，很多研究者和开发者在选择框架时可能会举棋不定。基于这种情况，grid.ai CEO、纽约大学博士 William Falcon 创建了 PyTorch Lightning，为 PyTorch 披上了一件 Keras 的外衣。

PyTorch 已足够简单易用，但简单易用不等于方便快捷。特别是做大量实验时，很多东西都会变得复杂，代码也会变得庞大，此时就容易出错。针对该问题，就有了 PyTorch Lightning。它可以重构你的 PyTorch 代码，抽出复杂重复部分，让你专注于核心的构建，让你的实验更快速更便捷地开展迭代。

为使代码能够用 Lightning 运行，本文将简要介绍如何 将 PyTorch 组织到 Lightning 中

一、简介本指南将展示如何分两步将 PyTorch 代码组织成 Lightning。使用 PyTorch Lightning 组织代码，可以使代码：保留所有灵活性（这全是纯 PyTorch），但去除了大量样板（boilerplate） 将研究代码与工程解耦，更具可读性 更容易复现 通过自动化大多数训练循环和棘手的工程设计，减少了出错的可能性 可扩展到任何硬件而无需更改模型二、安装PyTorch Lightning可用 pip 安装pip install pytorch-...