Pytorch-lightning学习
pytorch-lightning1.pytorch-lightning总览将pytorch主要训练过程封装起来 pytorch 和 pl 本质上代码是完全相同的。只不过pytorch需要自己造轮子（如model, dataloader, loss, train，test，checkpoint, save model等等都需要自己写），而pl 把这些模块都结构化了（类似keras）。 下面的例子很清晰的表明了两者的区别： lighting model 是跨硬件的，因此可以移除所有cuda()和device 2.模型搭建：2.1定义Lightning Module：class LitAutoEncoder(pl.LightningModule): def __init__(self): super().__init__() self.encoder = nn.Sequential( nn.Linear(28*28, 64), nn.ReLU(), nn.Linear(64, 3) ...