溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
模型的恢復
對于的模型的恢復來說，需要首先恢復模型的整個圖文件，之后從圖文件中讀取相應的節點信息。
存儲的模型文件包括四個子文件，如下：
現在假如我想恢復模型中的某個節點信息：
需要注意的是在使用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恢復模型的特定值時出錯就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
