如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)
更新時(shí)間：2020年02月06日 14:50:13 作者：深度學(xué)習(xí)小學(xué)生
今天小編就為大家分享一篇如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
模型的恢復(fù)
對(duì)于的模型的恢復(fù)來(lái)說(shuō)，需要首先恢復(fù)模型的整個(gè)圖文件，之后從圖文件中讀取相應(yīng)的節(jié)點(diǎn)信息。
存儲(chǔ)的模型文件包括四個(gè)子文件，如下：
現(xiàn)在假如我想恢復(fù)模型中的某個(gè)節(jié)點(diǎn)信息：
需要注意的是在使用saver.restore恢復(fù)權(quán)值的時(shí)候，參數(shù)的設(shè)置需要萬(wàn)分注意：
# 先加載圖文件
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)文章
使用Django和Flask獲取訪問(wèn)來(lái)源referrer
這篇文章主要介紹了使用Django和Flask獲取訪問(wèn)來(lái)源referrer，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04
這篇文章主要介紹了Python大文件排序的方法,涉及Python針對(duì)文件、緩存及日期等操作的相關(guān)技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-07-07
python深度學(xué)習(xí)之多標(biāo)簽分類器及pytorch實(shí)現(xiàn)源碼
這篇文章主要為大家介紹了python深度學(xué)習(xí)之多標(biāo)簽分類器的使用說(shuō)明及pytorch的實(shí)現(xiàn)源碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助2022-01-01
這篇文章主要介紹了python在控制臺(tái)輸出進(jìn)度條的方法,實(shí)例分析了Python輸出進(jìn)度條效果的方法,需要的朋友可以參考下2015-06-06
這篇文章主要介紹了Python 導(dǎo)入文件過(guò)程圖解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10
Python數(shù)據(jù)分析?Pandas?Series對(duì)象操作
這篇文章主要介紹了Python數(shù)據(jù)分析之Pandas?Series對(duì)象,文章基于python的相關(guān)資料展開(kāi)詳細(xì)內(nèi)容介紹，具有一定的參考價(jià)值，需要的小伙伴可以參考一下2022-05-05
使用Python的SymPy庫(kù)解決數(shù)學(xué)運(yùn)算問(wèn)題的方法
這篇文章主要介紹了使用Python的SymPy庫(kù)解決數(shù)學(xué)運(yùn)算問(wèn)題的方法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-03-03
這篇文章主要為大家詳細(xì)介紹了python腳本定時(shí)發(fā)送郵件，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-12-12
我已經(jīng)使用Python編程有多年了，即使今天我仍然驚奇于這種語(yǔ)言所能讓代碼表現(xiàn)出的整潔和對(duì)DRY編程原則的適用。這些年來(lái)的經(jīng)歷讓我學(xué)到了很多的小技巧和知識(shí)，大多數(shù)是通過(guò)閱讀很流行的開(kāi)源軟件，如Django, Flask, Requests中獲得的2013-01-01
python關(guān)于調(diào)用函數(shù)外的變量實(shí)例
今天小編就為大家分享一篇python關(guān)于調(diào)用函數(shù)外的變量實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12