tensorflow 重置/清除計算圖的實現
阿新 • • 發佈:2020-01-20
呼叫tf.reset_default_graph()重置計算圖
當在搭建網路檢視計算圖時,如果重複執行程式會導致重定義報錯。為了可以在同一個執行緒或者互動式環境中(ipython/jupyter)重複除錯計算圖,就需要使用這個函式來重置計算圖,隨後修改計算圖再次執行。
#重置計算圖,清理當前定義節點 import tensorflow as tf tf.reset_default_graph() #Your model defined below #
需要注意的是,下面三種情況使用這個函式會報錯:
#1 with graph.as_default(): #不能用 #2 with tf.Session(): block. #不能用 #3 tf.InteractiveSession() #Your regions #不能用 sess.close().
也就是說這個函式需要在with tf.session()外部呼叫。
以上這篇tensorflow 重置/清除計算圖的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。