欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

如何解決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)文章

最新評(píng)論