我有一个Inception_resnet_v2模型来提取特征。模型的速度大约是3s,这对于我们的目标来说太慢了。我使用在URL(https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/model_pruning)中发布的修剪方法layer.masked_conv2d来修剪我们的模型。最后,按照步骤s ,我采用strip_pruning_vars从模型中删除修剪操作,但是修剪模型中的masked_weight矩阵是稀疏矩阵,由0和实际权重组成,这意味着计算复杂度与以前的mdoel相同。此外,我已经测试了修剪模型,该模型花费> 3s来预测特征。是否有人遇到相同的问题,以及如何解决?
掩码代码
PlotlyViaWindowModule