2初級技術工具
PyTorch Lightning (訓練框架)
PyTorch Lightning
建於 PyTorch (Torch Compile) 上的高階訓練框架，以 Trainer 與 LightningModule 減少樣板程式碼，支援多 圖形處理單元 (GPU / 圖形處理器)、混合精度與 機器學習運維 整合。
詳細解釋
PyTorch Lightning 將 PyTorch (Torch Compile) 的訓練迴圈、分散式、 checkpoint、日誌等抽象成 Trainer 與 LightningModule，開發者只需定義 forward、training_step 等，即可獲得多 GPU/TPU、混合精度、早停與與 TensorBoard/機器學習運維 工具整合。降低樣板程式碼並利於重現與部署。與 PyTorch (Torch Compile)、機器學習運維、梯度檢查點 (Gradient Checkpointing) 相關。