如何解決tensorflow恢復模型的特定值時出錯

更新時間：2020年02月06日 14:50:13 作者：深度學習小學生

今天小編就為大家分享一篇如何解決tensorflow恢復模型的特定值時出錯，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

模型的恢復

對于的模型的恢復來說，需要首先恢復模型的整個圖文件，之后從圖文件中讀取相應的節(jié)點信息。

存儲的模型文件包括四個子文件，如下：

現(xiàn)在假如我想恢復模型中的某個節(jié)點信息：

需要注意的是在使用saver.restore恢復權(quán)值的時候，參數(shù)的設(shè)置需要萬分注意：

# 先加載圖文件
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:
  # 加載權(quán)值
  saver.restore(sess, "./model/save_model") # 這里才是重點，注意使用的是save_model，而不是save_model.data-00000-of-00001，不要帶一串后綴，如果使用后者的話，會報錯！?。。?！
  print(sess.run(a_val))

以上這篇如何解決tensorflow恢復模型的特定值時出錯就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

使用Django和Flask獲取訪問來源referrer
這篇文章主要介紹了使用Django和Flask獲取訪問來源referrer，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2021-04-04
Python實現(xiàn)大文件排序的方法
這篇文章主要介紹了Python大文件排序的方法,涉及Python針對文件、緩存及日期等操作的相關(guān)技巧,具有一定參考借鑒價值,需要的朋友可以參考下
2015-07-07
python深度學習之多標簽分類器及pytorch實現(xiàn)源碼
這篇文章主要為大家介紹了python深度學習之多標簽分類器的使用說明及pytorch的實現(xiàn)源碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助
2022-01-01
python在控制臺輸出進度條的方法
這篇文章主要介紹了python在控制臺輸出進度條的方法,實例分析了Python輸出進度條效果的方法,需要的朋友可以參考下
2015-06-06
Python 導入文件過程圖解
這篇文章主要介紹了Python 導入文件過程圖解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2019-10-10
Python數(shù)據(jù)分析?Pandas?Series對象操作
這篇文章主要介紹了Python數(shù)據(jù)分析之Pandas?Series對象,文章基于python的相關(guān)資料展開詳細內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下
2022-05-05
使用Python的SymPy庫解決數(shù)學運算問題的方法
這篇文章主要介紹了使用Python的SymPy庫解決數(shù)學運算問題的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2019-03-03
python腳本定時發(fā)送郵件
這篇文章主要為大家詳細介紹了python腳本定時發(fā)送郵件，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2020-12-12
Python高效編程技巧
我已經(jīng)使用Python編程有多年了，即使今天我仍然驚奇于這種語言所能讓代碼表現(xiàn)出的整潔和對DRY編程原則的適用。這些年來的經(jīng)歷讓我學到了很多的小技巧和知識，大多數(shù)是通過閱讀很流行的開源軟件，如Django, Flask, Requests中獲得的
2013-01-01
python關(guān)于調(diào)用函數(shù)外的變量實例
今天小編就為大家分享一篇python關(guān)于調(diào)用函數(shù)外的變量實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-12-12