[发明专利]一种4比特量化推理的方法、装置、设备及可读介质在审
申请号: | 202010980722.3 | 申请日: | 2020-09-17 |
公开(公告)号: | CN112200311A | 公开(公告)日: | 2021-01-08 |
发明(设计)人: | 王曦辉 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06N3/063 | 分类号: | G06N3/063;G06N3/08;G06N5/04 |
代理公司: | 北京连和连知识产权代理有限公司 11278 | 代理人: | 刘小峰;陈黎明 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种4比特量化推理的方法,包括以下步骤:训练生成伪量化模型,将伪量化模型中的归一化层合并入卷积层;对权重参数的数据类型进行等价变换和再量化,以将伪量化模型转换为量化模型;基于量化模型将常量折叠合并以生成输出数据类型为int4的推理模型;以及基于推理模型进行推理计算。本发明还公开了一种4比特量化推理的装置、计算机设备和可读存储介质。本发明实现GPU上的4比特推理,避免CPU中缺少int4这种数据类型而导致数据无法存储的问题,同时使模型体积为原始模型的1/8,推理时占用的内存也会减少,变为原来的1/8,大大加速了推理计算的过程。 | ||
搜索关键词: | 一种 比特 量化 推理 方法 装置 设备 可读 介质 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202010980722.3/,转载请声明来源钻瓜专利网。