如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)
更新時(shí)間：2020年02月06日 14:50:13 作者：深度學(xué)習(xí)小學(xué)生
今天小編就為大家分享一篇如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
模型的恢復(fù)
對(duì)于的模型的恢復(fù)來說，需要首先恢復(fù)模型的整個(gè)圖文件，之后從圖文件中讀取相應(yīng)的節(jié)點(diǎn)信息。
存儲(chǔ)的模型文件包括四個(gè)子文件，如下：
現(xiàn)在假如我想恢復(fù)模型中的某個(gè)節(jié)點(diǎn)信息：
需要注意的是在使用saver.restore恢復(fù)權(quán)值的時(shí)候，參數(shù)的設(shè)置需要萬分注意：
# 先加載圖文件
saver = tf.train.import_meta_graph("./model/save_model.meta")
graph = tf.get_default_graph()
# 其中的一個(gè)節(jié)點(diǎn)
a_val = graph.get_tensor_by_name("var/a_val:0") # 這里是我的節(jié)點(diǎn)名稱，這都不是重點(diǎn)
with tf.Session() as sess:
# 加載權(quán)值
saver.restore(sess, "./model/save_model") # 這里才是重點(diǎn)，注意使用的是save_model，而不是save_model.data-00000-of-00001，不要帶一串后綴，如果使用后者的話，會(huì)報(bào)錯(cuò)?。。。?！
print(sess.run(a_val))
以上這篇如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python庫coala代碼分析和自動(dòng)化檢查改進(jìn)工具使用探索
Python的coala工具是一個(gè)強(qiáng)大的代碼分析和自動(dòng)化改進(jìn)工具,它可以幫助開發(fā)人員自動(dòng)化地檢查代碼并提供改進(jìn)建議,本文將詳細(xì)介紹coala的功能和用法,并提供豐富的示例代碼來幫助大家深入了解它2024-01-01
python opencv3實(shí)現(xiàn)人臉識(shí)別(windows)
這篇文章主要為大家詳細(xì)介紹了python opencv3實(shí)現(xiàn)人臉識(shí)別程序，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-05-05
今天小編就為大家分享一篇基于pytorch的保存和加載模型參數(shù)的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-08-08
python開發(fā)中range()函數(shù)用法實(shí)例分析
這篇文章主要介紹了python開發(fā)中range()函數(shù)用法,以實(shí)例形式較為詳細(xì)的分析了Python中range()函數(shù)遍歷列表的相關(guān)技巧,需要的朋友可以參考下2015-11-11
基于opencv和pillow實(shí)現(xiàn)人臉識(shí)別系統(tǒng)(附demo)
人臉識(shí)別就是一個(gè)程序能識(shí)別給定圖像或視頻中的人臉，本文主要介紹了opencv和pillow實(shí)現(xiàn)人臉識(shí)別系統(tǒng)，本文不涉及分類器、訓(xùn)練識(shí)別器等算法原理，感興趣的可以了解一下2021-11-11
Python json模塊與jsonpath模塊區(qū)別詳解
這篇文章主要介紹了Python json模塊與jsonpath模塊區(qū)別詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
IPython庫中的display函數(shù)的簡(jiǎn)介、使用方法、應(yīng)用案例詳細(xì)攻略
display 函數(shù)可以接受一個(gè)或多個(gè)參數(shù)，每個(gè)參數(shù)都是一個(gè) Python 對(duì)象。它會(huì)自動(dòng)根據(jù)對(duì)象的類型選擇合適的顯示方式，并在 Jupyter Notebook 中顯示出來，這篇文章主要介紹了IPython庫中的display函數(shù)的簡(jiǎn)介、使用方法、應(yīng)用案例詳細(xì)攻略,需要的朋友可以參考下2023-04-04
opencv python 對(duì)指針儀表讀數(shù)識(shí)別的兩種方式
這篇文章主要介紹了opencv python 對(duì)指針儀表讀數(shù)識(shí)別的兩種方式,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-01-01
Python中.join()和os.path.join()兩個(gè)函數(shù)的用法詳解
join()是連接字符串?dāng)?shù)組而os.path.join()是將多個(gè)路徑組合后返回。接下來通過本文重點(diǎn)給大家介紹Python中.join()和os.path.join()兩個(gè)函數(shù)的用法，感興趣的朋友一起看看吧2018-06-06
Pyramid Mako模板引入helper對(duì)象的步驟方法
ylons中的mako模板，默認(rèn)會(huì)引入一個(gè)helper對(duì)象，我們可以在里面擴(kuò)展方法，應(yīng)對(duì)在模板輸出時(shí)候會(huì)常用的操作，那么在Pyramid中如何默認(rèn)引入同樣的輔助類到模板中2013-11-11