本发明公开了一种联合比特量化和比特权重搜索的模型压缩方法,包括以下步骤：S1、加载图像数据并对图像数据进行预处理；S2、建立卷积神经网络模型,设定卷积神经网络模型中对应的比特搜索空间和权重搜索空间；S3、对卷积神经网络模型进行联合搜索训练；S4、选取比特搜索结果中的最大概率位宽,以构建轻量化网络,并保存联合搜索时最后以最大概率位宽得到的权重参数作为轻量化网络的权重参数,再进行权重搜索来进一步优化轻量化网络模型,输出压缩了的优化轻量化网络模型。
科研之友
微信
新浪微博
Facebook
分享链接
