TensorFlow函数：tf.graph_util.remove_training_nodes

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函数删除了不必要的节点的列表.