0
点赞
收藏
分享

微信扫一扫

Pytorch Lightning【在Pytorch基础上进行封装的库,为了让用户能够脱离PyTorch一些繁琐的细节,专注于核心代码的构建,提供了许多实用工具,可以让实验更加高效】

PyTorch非常易于使用,可以构建复杂的AI模型。但是一旦研究变得复杂,并且将诸如多GPU训练,16位精度和TPU训练之类的东西混在一起,用户很可能会引入错误。

PyTorch Lightning完全解决了这个问题:

  • Lightning会构建您的PyTorch代码,以便可以抽象出训练的细节。这使得AI研究可扩展且可快速迭代。
  • Pytorch Lightning是在Pytorch基础上进行封装的库,为了让用户能够脱离PyTorch一些繁琐的细节,专注于核心代码的构建,提供了许多实用工具,可以让实验更加高效。

PyTorch Lightning中提供了以下比较方便的功能:

  • multi-GPU训练
  • 半精度训练
  • TPU 训练
  • 将训练细节进行抽象,从而可以快速迭代



参考资料:
pytorch_lightning 全程笔记
Pytorch-Lightning基本方法介绍
人工智能,从PyTorch到PyTorch Lightning简要介绍

举报

相关推荐

0 条评论