開始使用免費開始

LightningModule 的用途

LightningModule 是 PyTorch Lightning 在結構化深度學習專案時的核心。它集中處理模型架構、forward 傳遞與訓練邏輯,並將訓練迴圈與裝置設定的協調工作留給 Trainer。

以下哪一個敘述最能說明 LightningModule 在 PyTorch Lightning 中的主要角色?

本練習屬於課程

Scalable AI Models with PyTorch Lightning

檢視課程

動手互動練習

將理論付諸實踐,立即體驗我們的互動練習

開始練習