模型的恢復

對於的模型的恢復來說，需要首先恢復模型的整個圖檔案，之後從圖檔案中讀取相應的節點資訊。

儲存的模型檔案包括四個子檔案，如下：

現在假如我想恢復模型中的某個節點資訊：

需要注意的是在使用saver.restore恢復權值的時候，引數的設定需要萬分注意：

# 先載入圖檔案
saver = tf.train.import_meta_graph("./model/save_model.meta")
graph = tf.get_default_graph()
# 其中的一個節點
a_val = graph.get_tensor_by_name("var/a_val:0") # 這裡是我的節點名稱，這都不是重點

with tf.Session() as sess:
  # 載入權值
  saver.restore(sess,"./model/save_model") # 這裡才是重點，注意使用的是save_model，而不是save_model.data-00000-of-00001，不要帶一串字尾，如果使用後者的話，會報錯！！！！！
  print(sess.run(a_val))

以上這篇如何解決tensorflow恢復模型的特定值時出錯就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。