会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
LyleChen
博客园
首页
新随笔
联系
订阅
管理
2020年8月9日
Pytorch原生AMP支持使用方法(1.6版本)
摘要: AMP:Automatic mixed precision,自动混合精度,可以在神经网络推理过程中,针对不同的层,采用不同的数据精度进行计算,从而实现节省显存和加快速度的目的。 在Pytorch 1.5版本及以前,通过NVIDIA出品的插件apex,可以实现amp功能。 从Pytorch 1.6版本
阅读全文
posted @ 2020-08-09 22:49 LyleChen
阅读(5090)
评论(0)
推荐(2)
编辑
公告