【PyTorch的快速研究框架】Lightning是PyTorch上非常轻量级的包装器。它将核心培训和验证逻辑推迟给您,并使其余部分自动化。它保证了核心培训逻辑的正确,现代最佳实践。在开始一个新项目时,您要做的最后一件事是重新编码训练循环,多集群训练,16位精度,早期停止,模型加载/保存,何时验证等等…您很可能花费很长时间来解决所有的问题,甚至没有进入研究的核心。借助Lightning,您可以保证代码的这些部分能够正常工作,这样您就可以专注于研究的内容:数据和培训/验证循环逻辑。
https://github.com/williamFalcon/pytorch-lightning