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

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

 更新時間：2020年02月06日 14:50:13   作者：深度學習小學生  
今天小編就為大家分享一篇如何解決tensorflow恢復模型的特定值時出錯，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

模型的恢復

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

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

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

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

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

相關文章

最新評論