PyTorch Lightning入門(mén)教程 #2 - Lightning Mod
2023-07-11 16:12 作者:九轉(zhuǎn)大腸制作 | 我要投稿
copilt 真好用，自動(dòng)補(bǔ)全代碼
==========
class NN(pl.LightningModule):
重新繼承 pl.LightningModule，和 后面的Trainer集成。
所有功能函數(shù)寫(xiě)在類里面了，這些應(yīng)該 是重載 pl.LightningModule 中的 函數(shù)?？梢渣c(diǎn)進(jìn)去 進(jìn)一步查看！
標(biāo)簽：
