Tensorflow:如何在急切模式下运行基于图的检查点以及如何将急切模型转换为基于图

时间:2018-08-09 11:35:02

标签: python tensorflow

这是我在Tensorflow中需要的

  1. 加载基于图形的预训练VGG检查点(或.pb)模型
  2. 以急切的执行模式修剪并微调模型
  3. 再次将其转换为.pb文件。

为什么

  1. 大多数预训练模型现在都是基于张量流的图,因为渴望执行是一种新工具。
  2. 我知道tf.contrib.model_pruning可用于修剪神经网络,但是,它仅支持基于幅度的修剪。我可以在卷积层中修剪整个过滤器(或通道),或者在密集层中删除节点,显然动态模型更好,所以我想在急切的执行模式中修剪。
  3. 基于图的模型适合部署,因此我想再次将其转换为.pb模型

有人知道答案吗?非常感谢〜

0 个答案:

没有答案