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