实战对比PyTorch VS PyTorch Lighting以MNIST为例最近正在准备毕业设计的相关内容,发现其使用的是PyTorch Lightning和之前的PyTorch有一定的差距,因此就主动了解了它们之间的差距。 如果你用过PyTorch,一定会经历过类似的场景:手动编写训练循环、反复切换model.train()和model.eval()模式、手动管理GPU设备、写一堆日志记录逻辑……这些重复的工程代码占用了大量时间,却与核心的模型研究无关。 PyTorch Lightning正是为了解决这些问题而诞生的。它不是替代PyTorch的新框架,而是在PyTorch之上构