1. 程式人生 > 程式設計 >tensorflow 重置/清除計算圖的實現

tensorflow 重置/清除計算圖的實現

呼叫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 重置/清除計算圖的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。