阅读(11.4k) 书签 (0)

TensorFlow函数:tf.graph_util.remove_training_nodes

2018-05-16 10:56 更新

tf.graph_util.remove_training_nodes函数

tf.graph_util.remove_training_nodes(
    input_graph,
    protected_nodes=None
)

定义在:tensorflow/python/framework/graph_util_impl.py.

删除不需要进行推理(inference)的节点.

在训练过程中,有一些节点,比如Identity和checknerics,它们只在训练时有用,并且可以在仅用于推理的图形中删除.我们识别并删除它们,返回一个等效图.具体而言,CheckNumerics节点总是被删除,并且没有涉及控制边缘的Identity节点被拼接出来,以便它们的输入和输出直接连接.

参数:

  • input_graph:需要分析和删除的模型.
  • protected_nodes:无条件保留的节点名称的可选列表.这对于保留标识输出节点非常有用.

返回值:

tf.graph_util.remove_training_nodes函数删除了不必要的节点的列表.