CVPR | 引入Apex技术:PyTorch扩展其工具包

作者: CBISMB

责任编辑: 贾西贝

来源: ISMB

时间: 2018-06-22 09:51

关键字: NVIDIA,Apex,CVPR

浏览: 0

点赞: 0

收藏: 0

在计算机视觉与模式识别大会 (CVPR) 上,NVIDIA率先展示了Apex技术的早期版本——这是一款帮助用户利用NVIDIA Volta GPU实现深度学习训练性能最大化的开源式PyTorch扩展。

在转换网络、情感分析和图像分类中的先进混合精度训练的启发下,NVIDIA PyTorch开发者创造出了一些新工具,并将这些方法提供给所有级别的PyTorch用户。

Apex技术中的混合精度工具包能够在提高训练速度的同时,保持单精度训练的准确性和稳定性。

具体来说,Apex技术可自动执行对FP16 矩阵或 FP32矩阵的维护操作、自动处理主参数转换并且引入了自动损失缩放功能,在更改不超过4行现有代码的情况下即可使用。

安装要求具备CUDA 9、PyTorch 0.4或更高版本以及Python 3。模块和工具包仍在开发进程中,我们期待您的反馈,以帮助我们把这些工具包做得更好。

您可以下载代码并通过学习文档及教程开始使用,我们的GitHub 页面提供了展示案例。

©本站发布的所有内容,包括但不限于文字、图片、音频、视频、图表、标志、标识、广告、商标、商号、域名、软件、程序等,除特别标明外,均来源于网络或用户投稿,版权归原作者或原出处所有。我们致力于保护原作者版权,若涉及版权问题,请及时联系我们进行处理。