溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)

發(fā)布時(shí)間：2020-08-25 11:33:07 來(lái)源：腳本之家 閱讀：126 作者：深度學(xué)習(xí)小學(xué)生 欄目：開(kāi)發(fā)技術(shù)

模型的恢復(fù)

對(duì)于的模型的恢復(fù)來(lái)說(shuō)，需要首先恢復(fù)模型的整個(gè)圖文件，之后從圖文件中讀取相應(yīng)的節(jié)點(diǎn)信息。

存儲(chǔ)的模型文件包括四個(gè)子文件，如下： 

如何解決tensorflow恢復(fù)模型的特定值時(shí)出錯(cuò)

現(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è)參考，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手