【Pytorch Lightning】Pytorch Lightning 初入

2022/3/1 6:24:50

本文主要是介绍【Pytorch Lightning】Pytorch Lightning 初入,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

背景

看到这个,我相信你对Pytorch Lightning 有一定了解。虽然Pytorch已经很好了,但是在工程上,论文复现上等等,我们有时需要根据论文快速复现模型、有时有了好的idea想快速实现、有时工程上需要不断调优等等。这时我们就需要一个工具,能够快速使用他去实现模型,调优等。

前段时间这个工具就火了,就是Pytorch Lightning:官方地址.Github上已经17k+了,社区也很活跃。强大的工具也同时伴随着较多的学习成本。后面就根据官方文档深入了解这个工具。

当然,官方文档写的是比较好的,能耐得住看官方文档更好。后面的一系列文章就可以忽略了。

在了解这个工具之前,是需要有一定模型构建,调优的一些基础,Pytorch基础等,切记!!!

两步了解Lightning

使用PyTorch Lightning之后,会让你的代码有如下特点:

  • 代码依然很灵活,但移除了大量的冗余代码
  • 通过将研究代码与工程分离,使其更具可读性
  • 容易复用
  • 通过自动化大部分训练循环和复杂的工程,减少了出错的可能性
  • 不改变模型也能够适应任何机器

看一下官方视频,了解一下基本操作:



这篇关于【Pytorch Lightning】Pytorch Lightning 初入的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程