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.

刪除不需要進(jìn)行推理(inference)的節(jié)點(diǎn).

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

參數(shù)：

• input_graph：需要分析和刪除的模型.
• protected_nodes：無(wú)條件保留的節(jié)點(diǎn)名稱(chēng)的可選列表.這對(duì)于保留標(biāo)識(shí)輸出節(jié)點(diǎn)非常有用.

返回值：

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