我已经使用了条带修剪vars.py修剪细长模型(.pb),但是修剪模型的速度比未修剪模型慢

时间:2019-08-21 05:32:01

标签: python-3.x tensorflow model slim pruning

我有一个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

0 个答案:

没有答案