更轻量的Pytorch来了

设为“星标”，重磅干货，第一时间送达。

这是阿三的第 97 期分享 

作者 | 阿三

首发 | 程序员遇见GitHub

大家好，我是阿三，今天给大家带来的是最近很火的Pytorch Lighting。

一.Pytorch Lightng

Pytorch在最近几年已经成为了最受欢迎的深度学习工具，但是当系统比较复杂时，就容易出现很多错误，于是Pytorch Lighting应运而生。

Pytorch Lighting的作者在自己的github里提到，Lightning是一种组织PyTorch代码的方法，可以将科学代码与工程代码解耦。它更像是一个PyTorch风格指南，而不是一个框架。

它将代码组织成3个不同的类别:

研究代码，工程代码和不重要的代码

下面是一个例子：

剩余代码由trainer自动完成。

相当于我们只需要组织pytorch代码，不需要抽象，trainer能够帮我们自动抽象出来，然后我们可以重写任何想做的事情，比如实现自己的分布式训练、16位精度。不用担心GPU，TPU等问题，trainer已经全部处理好。

class LitModel(LightningModule):
def optimizer_step(self, current_epoch, batch_idx, optimizer, optimizer_idx,
second_order_closure=None, on_tpu=False, using_native_amp=False, using_lbfgs=False):
optimizer.step()

def optimizer_zero_grad(self, current_epoch, batch_idx, optimizer, opt_idx):
optimizer.zero_grad()

github地址：

https://github.com/PyTorchLightning/pytorch-lightning

 二 希望和大家有互动

阿三在写作过程中希望和大家进行互动，具体形式，大家可以用打卡形式进行留言，内容可以是对这篇文章看法，和希望阿三在分享什么。形式如下“

打卡+第几天+内容，比如：

打卡+第一天+希望明天分享机器学习入门的知识。

从6月18日开始，新的一轮打卡已经开始！请各位小伙伴留意！

为了鼓励大家反馈，阿三也有一定的反馈，对于打卡满30天的童鞋前五名（前几名代表最早开始在后台告知阿三的满足要求童鞋，如果有6名童鞋满足要求，但是只给最早5个告诉阿三已经打满30天，截图为证，后台截图发送告知），阿三送出128元的红包~

阿三谢谢大家支持！一起做好内容，能对大家有一点帮助是阿三最大的心愿。