如何解決tensorflow恢復模型的特定值時出錯
更新時間:2020年02月06日 14:50:13 作者:深度學習小學生
今天小編就為大家分享一篇如何解決tensorflow恢復模型的特定值時出錯,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
模型的恢復
對于的模型的恢復來說,需要首先恢復模型的整個圖文件,之后從圖文件中讀取相應的節(jié)點信息。
存儲的模型文件包括四個子文件,如下:
現在假如我想恢復模型中的某個節(jié)點信息:
需要注意的是在使用saver.restore恢復權值的時候,參數的設置需要萬分注意:
# 先加載圖文件 saver = tf.train.import_meta_graph("./model/save_model.meta") graph = tf.get_default_graph() # 其中的一個節(jié)點 a_val = graph.get_tensor_by_name("var/a_val:0") # 這里是我的節(jié)點名稱,這都不是重點 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恢復模型的特定值時出錯就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python庫coala代碼分析和自動化檢查改進工具使用探索
Python的coala工具是一個強大的代碼分析和自動化改進工具,它可以幫助開發(fā)人員自動化地檢查代碼并提供改進建議,本文將詳細介紹coala的功能和用法,并提供豐富的示例代碼來幫助大家深入了解它2024-01-01基于opencv和pillow實現人臉識別系統(tǒng)(附demo)
人臉識別就是一個程序能識別給定圖像或視頻中的人臉,本文主要介紹了opencv和pillow實現人臉識別系統(tǒng),本文不涉及分類器、訓練識別器等算法原理,感興趣的可以了解一下2021-11-11Python json模塊與jsonpath模塊區(qū)別詳解
這篇文章主要介紹了Python json模塊與jsonpath模塊區(qū)別詳解,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-03-03IPython庫中的display函數的簡介、使用方法、應用案例詳細攻略
display 函數可以接受一個或多個參數,每個參數都是一個 Python 對象。它會自動根據對象的類型選擇合適的顯示方式,并在 Jupyter Notebook 中顯示出來,這篇文章主要介紹了IPython庫中的display函數的簡介、使用方法、應用案例詳細攻略,需要的朋友可以參考下2023-04-04Python中.join()和os.path.join()兩個函數的用法詳解
join()是連接字符串數組而os.path.join()是將多個路徑組合后返回。接下來通過本文重點給大家介紹Python中.join()和os.path.join()兩個函數的用法,感興趣的朋友一起看看吧2018-06-06