权重剪枝后的模型如何加载到原网络中?

在网络剪枝部分,对核内一些不重要的权重置为“0”,保存下的模型大小还是没有变,分析原因是因为虽然将部分权重置0,但“0”权重还是占内存。于是我将“0”权重删掉,保存下来的权重大小确实减小了,但带来了新的问题,保存下的模型无法加载到原网络中进行推理,这是由于形状发生了改变,无法加载到网络。那么我应该如何将剪枝后的模型加载到原网络中呢?

你好,请问你找到解决方案了吗?

权重剪枝后的模型加载到原网络中