使用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
标签:无需,模型,lightning,pytorch,https,GPU,漂亮 来源: https://www.cnblogs.com/gczr/p/14417175.html