使用pytorch-lightning漂亮地進行深度學習研究(轉)


一、設計哲學

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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM