神经网络剪枝方法、装置、设备及存储介质
摘要:
本申请实施例公开了一种神经网络剪枝方法、装置、设备及存储介质,其包括:向原始网络模型和复制得到的剪枝网络模型输入同一心电信号,原始网络模型包括至少一个第一特征提取模块,剪枝网络模型包括至少一个第二特征提取模块;利用第二特征提取模块的第四卷积层中一通道内使用的卷积核参数处理第三卷积层输出的第一特征图对应的通道,利用处理后得到的第三特征图和第一特征提取模块的第二卷积层输出的第二特征图确定第二卷积层中对应通道的通道性能;根据通道性能确定第二卷积层中的待剪枝通道;根据待剪枝通道对第一卷积层和第二卷积层进行剪枝。采用上述方法可解决提高神经网络模型的性能时增加结构复杂度引起的推理时间变长的技术问题。
0/0