PyTorch Lightning入门教程 #2 - Lightning Mod

copilt 真好用，自动补全代码

==========

class NN(pl.LightningModule):

重新继承 pl.LightningModule，和 后面的Trainer集成。

所有功能函数写在类里面了，这些应该 是重载 pl.LightningModule 中的 函数。可以点进去 进一步查看！