我是h1

 

地址: 深圳湾科技生态园5栋D座1302室
客服电话:
4008400488
邮箱: bd@rainxn.com

墨客星球

友情链接

© 2019 深圳雨本信息技术有限公司 | 粤ICP备17123203号 网站建设:中企动力 深圳

联系我们

最新文章

拥抱数据浪潮,引领未来风向|2019大数据产业峰会圆满落幕
拥抱数据浪潮,引领未来风向|2019大数据产业峰会圆满落幕
第十一届中国深圳创新创业大赛盐田区预选赛暨2019人工智能创新创业大赛即将起航
第十一届中国深圳创新创业大赛盐田区预选赛暨2019人工智能创新创业大赛即将起航
AI数钢筋,人工智能的下一个蓝海应用
AI数钢筋,人工智能的下一个蓝海应用
人机协同,人工智能正扛起赋能医疗的大旗
人机协同,人工智能正扛起赋能医疗的大旗
AI产业现状:深度学习并没有你想的那么强大
AI产业现状:深度学习并没有你想的那么强大
曾经最被看好的独角兽,为什么成了中国第一家倒下的无人驾驶公司
曾经最被看好的独角兽,为什么成了中国第一家倒下的无人驾驶公司

TensorFlow 中的自动混合精度,可在 NVIDIA GPU 上实现更快的 AI 训练

快讯
2019年3月20日
浏览量
【摘要】:
TensorFlow技术博客发布新文,文章讨论道,混合精度训练利用半精度加速训练,在某些情况下即便是使用相同超参数进行单精度训练,也能达到相同的精度。如此一来,内存需求会减少。使用混合精度需要两个步骤:移植模型以在适当时候使用半精度数据类型;并使用损耗缩放来保留小的梯度值。我们(指英伟达)为TensorFlow引入了自动混合精度功能(现在可以在1.x中使用,很快也能在2.x中使用),这能通过英伟达

TensorFlow 技术博客发布新文,文章讨论道,混合精度训练利用半精度加速训练,在某些情况下即便是使用相同超参数进行单精度训练,也能达到相同的精度。如此一来,内存需求会减少。使用混合精度需要两个步骤:移植模型以在适当时候使用半精度数据类型;并使用损耗缩放来保留小的梯度值。我们(指英伟达)为 TensorFlow 引入了自动混合精度功能(现在可以在 1.x 中使用,很快也能在 2.x 中使用),这能通过英伟达的 Volta 和 Turing GPU 中的 Tensor Core 进行修改以提高训练性能。自动混合精度在 TensorFlow 中内部应用这两个步骤,在 NVIDIA 的 NGC 容器中使用单个环境变量,并在必要时实现更精细的控制。