文章/答案/技术大牛

发布

社区首页 >问答首页 >如何保存Tf.contrib模型剪枝？

问如何保存Tf.contrib模型剪枝？
EN

Stack Overflow用户

提问于 2018-07-21 12:40:00

回答 1查看 727关注 0票数 2

我已经建立了一个模型，并且我能够成功地使用tf.cont肋骨的模型剪枝模块，默认参数和稀疏度为90%，但问题是当我运行该模型时，它仍然需要与原始模型相同的执行时间，我的猜测是，tensorflow不是只运行被剪枝的版本，而是运行整个图的蒙面西部，这就是为什么即使在剪枝之后也没有改进。

那么，如何导出具有子图和相应权重的剪枝模型并使用它呢？

发布于 2018-08-27 17:45:58

strip_pruning_vars实用程序可能就是您要寻找的工具。

从训练图中移除剪枝操作一旦对模型进行了训练，就必须删除辅助变量(掩码、阈值)，并在上述步骤中修剪添加到图形中的操作。这可以使用strip_pruning_vars实用程序来实现。

你介意分享你的密码吗？

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/51456166

复制

相似问题

问如何保存Tf.contrib模型剪枝？EN