摘要
本申请提供了一种模型剪枝方法、装置、电子设备及存储介质;方法包括:获取待剪枝模型以及多个输入数据;通过目标子模型执行以下处理:对与目标子模型对应的输入数据进行预测,得到目标子模型的预测结果;定目标子模型对应的输入数据与其他输入数据之间的距离;基于每个子模型的预测结果,对目标子模型进行权重预测,得到目标子模型的预测权重,并融合距离以及预测权重,得到目标子模型的目标权重;基于每个子模型的目标权重,对待剪枝模型进行子模型剪枝,得到剪枝后的模型。通过本申请,能够提升子模型剪枝的准确性。
技术关键词
剪枝模型
模型剪枝方法
计算机可执行指令
样本
电子设备
强度
计算机程序产品
数据采集模块
处理器
可读存储介质
存储器
因子
参数
编码
数值