tensorflow恢復模型的特定值時出錯的問題解決

模型的恢復

對於的模型的恢復來說，需要首先恢復模型的整個圖文件，之後從圖文件中讀取相應的節點信息。
存儲的模型文件包括四個子文件，如下：

現在假如我想恢復模型中的某個節點信息：
需要注意的是在使用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))