TensorFlow函数:tf.graph_util.remove_training_nodes

由 Carrie 创建, 最后一次修改 2018-05-16

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

以上内容是否对您有帮助:
二维码
建议反馈
二维码