將變量從一個TensorFlow圖複製到另一個

Question

我有兩張張量圖。一個用於培訓，另一個用於評估。它們共享很多變量名稱。當我評估一個模型時，我想將火車圖中的所有變量值複製到測試圖中。顯然，我可以通過tf.train.Saver來完成，但是這個解決方案似乎並不適合我，尤其是我們必須爲此使用磁盤。

Answer 1

當你談論多個圖形，我假定你的意思是這樣的：

g1 = tf.Graph() 
with g1.as_default(): 
    # add your stuff 

g2 = tf.Graph() 
with g2.as_default(): 
    # add other stuff 

如果這是正確的，那麼你確定你真的需要兩個圖？你不能有一個由兩個連接組件組成的圖嗎？

使用多個圖形是discouraged（第47頁），因爲：

• 多個圖形需要多個會話，每個將嘗試在默認情況下使用所有可用的資源
• 無法通過它們之間的數據沒有通過python/numpy傳遞它們，這在分佈式中不起作用
• 在一個圖形內有較好的斷開子圖形

這也給你一個解決方案，如何在非分佈式設置中傳遞變量。