pytorch-lightning是基于pytorch的一个套壳项目,适配pytorch的版本同步更新速度很快。
它将训练的几个主要流程模块化,减少重复工作,同时让支持分布式训练,不同平台的训练迁移变得更加简单。 官网链接