使用pytorch-lightning漂亮地进行深度学习研究（转）

本文转载自查看原文 2021-02-19 17:26 453 pytorch

一、设计哲学

pytorch-lightning 是建立在pytorch之上的高层次模型接口。

pytorch-lightning 之于 pytorch，就如同keras之于 tensorflow。

通过使用 pytorch-lightning，用户无需编写自定义训练循环就可以非常简洁地在CPU、单GPU、多GPU、乃至多TPU上训练模型。

无需考虑模型和数据在cpu,cuda之间的移动，并且可以通过回调函数实现CheckPoint参数保存，实现断点续训功能。

实例参考：

使用pytorch-lightning漂亮地进行深度学习研究

https://github.com/PyTorchLightning/pytorch-lightning

https://pytorch-lightning.readthedocs.io/en/latest/index.html

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 Pytorch-Lightning基本方法介绍【深度学习框架】使用PyTorch进行数据处理 pytorch lightning使用（简要介绍） Pytorch——Pytorch Lightning框架使用手册【PyTorch深度学习】学习笔记之PyTorch与深度学习 PyTorch中使用深度学习（CNN和LSTM）的自动图像标题深度学习框架pytorch入门与实践（一）：torch的基本使用 [深度学习]pytorch部署之onnx 深度学习之入门Pytorch（1）------基础【深度学习】PyTorch之Squeeze()和Unsqueeze()