TensorFlow函數(shù)：tf.graph_util.remove_training_nodes

tf.graph_util.remove_training_nodes函數(shù)

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

定義在：tensorflow/python/framework/graph_util_impl.py.

刪除不需要進行推理(inference)的節(jié)點.

在訓練過程中,有一些節(jié)點,比如Identity和checknerics,它們只在訓練時有用,并且可以在僅用于推理的圖形中刪除.我們識別并刪除它們,返回一個等效圖.具體而言,CheckNumerics節(jié)點總是被刪除,并且沒有涉及控制邊緣的Identity節(jié)點被拼接出來,以便它們的輸入和輸出直接連接.

參數(shù)：

• input_graph：需要分析和刪除的模型.
• protected_nodes：無條件保留的節(jié)點名稱的可選列表.這對于保留標識輸出節(jié)點非常有用.

返回值：

tf.graph_util.remove_training_nodes函數(shù)刪除了不必要的節(jié)點的列表.