在 PyTorch Lightning 中,常用的回呼函式有 ModelCheckpoint 與 EarlyStopping。前者會依你選定的指標自動儲存表現最好的模型,後者則在改善趨於停滯時停止訓練,以避免過度擬合。來試試看你能否正確分類各回呼函式的重點敘述。
本練習屬於課程
將理論付諸實踐,立即體驗我們的互動練習
在本章中,我們將探討 PyTorch Lightning 如何簡化可擴充 AI 模型的開發與部署。從基礎概念開始,我們會走過 PyTorch Lightning 專案的核心結構,包括 LightningModule 與 Trainer 等必要元件,為更進階的 AI 解決方案打下扎實基礎。
本章將更深入 PyTorch Lightning,以高效率管理資料並優化模型訓練。我們會學習如何使用 LightningDataModule 建立模組化且可重用的資料工作流程,透過驗證與測試精準評估你的模型,並利用 Lightning Callbacks 強化訓練流程,自動化模型改進並避免過度擬合。
當前練習
學習讓深度學習模型更精簡、更迅速,以便實際部署。本章介紹動態量化、剪枝與 TorchScript 轉換等技術,協助你在不犧牲準確度的前提下,降低模型大小與延遲。