一、設計哲學
pytorch-lightning 是建立在pytorch之上的高層次模型接口。
pytorch-lightning 之於 pytorch,就如同keras之於 tensorflow。
通過使用 pytorch-lightning,用戶無需編寫自定義訓練循環就可以非常簡潔地在CPU、單GPU、多GPU、乃至多TPU上訓練模型。
無需考慮模型和數據在cpu,cuda之間的移動,並且可以通過回調函數實現CheckPoint參數保存,實現斷點續訓功能。
實例參考:
使用pytorch-lightning漂亮地進行深度學習研究
GitHub高贊!PyTorch Lightning 你值得擁有!
上手傳送門
https://github.com/PyTorchLightning/pytorch-lightning
https://pytorch-lightning.readthedocs.io/en/latest/index.html