看到这个,我相信你对Pytorch Lightning 有一定了解。虽然Pytorch已经很好了,但是在工程上,论文复现上等等,我们有时需要根据论文快速复现模型、有时有了好的idea想快速实现、有时工程上需要不断调优等等。这时我们就需要一个工具,能够快速使用他去实现模型,调优等。
前段时间这个工具就火了,就是Pytorch Lightning:官方地址.Github上已经17k+了,社区也很活跃。强大的工具也同时伴随着较多的学习成本。后面就根据官方文档深入了解这个工具。
当然,官方文档写的是比较好的,能耐得住看官方文档更好。后面的一系列文章就可以忽略了。
在了解这个工具之前,是需要有一定模型构建,调优的一些基础,Pytorch基础等,切记!!!
使用PyTorch Lightning之后,会让你的代码有如下特点:
看一下官方视频,了解一下基本操作: