火炬闪电模板
Pytorch-Lightning是一个很好的库,或者说是Pytorch的抽象和包装。它的好处是可替代性强,易维护,逻辑清晰等。缺点也很明显,这个包需要学习和理解的内容还是如果直接按照官方的模板写的代码,小型项目还好,如果是大型项目,有复数个需要调试验证的模型和数据集,那就不太好办,甚至更加麻烦了。经过几天的摸索和调试,我总结出了下面这样一套好的用的模板,也可以说是对Pytorch-Lightning的进一步摘要。
欢迎大家尝试这一套代码风格,如果用习惯的话还是相当方便替代的,也不