【Pytorch Lightning】Pytorch Lightning 初入
作者:互联网
背景
看到这个,我相信你对Pytorch Lightning 有一定了解。虽然Pytorch已经很好了,但是在工程上,论文复现上等等,我们有时需要根据论文快速复现模型、有时有了好的idea想快速实现、有时工程上需要不断调优等等。这时我们就需要一个工具,能够快速使用他去实现模型,调优等。
前段时间这个工具就火了,就是Pytorch Lightning:官方地址.Github上已经17k+了,社区也很活跃。强大的工具也同时伴随着较多的学习成本。后面就根据官方文档深入了解这个工具。
当然,官方文档写的是比较好的,能耐得住看官方文档更好。后面的一系列文章就可以忽略了。
在了解这个工具之前,是需要有一定模型构建,调优的一些基础,Pytorch基础等,切记!!!
两步了解Lightning
使用PyTorch Lightning之后,会让你的代码有如下特点:
- 代码依然很灵活,但移除了大量的冗余代码
- 通过将研究代码与工程分离,使其更具可读性
- 容易复用
- 通过自动化大部分训练循环和复杂的工程,减少了出错的可能性
- 不改变模型也能够适应任何机器
看一下官方视频,了解一下基本操作:
标签:代码,Lightning,初入,Pytorch,官方,文档,工具 来源: https://blog.csdn.net/meiqi0538/article/details/123172885