jlzzjlzz亚洲乱熟在线播放

系統城裝機大師 - 唯一官網:www.farandoo.com!

當前位置:首頁 > 腳本中心 > htc > 詳細頁面

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

時間:2020-02-06來源:系統城作者:電腦系統城

模型的恢復

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

存儲的模型文件包括四個子文件,如下: 

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

現在假如我想恢復模型中的某個節點信息:

需要注意的是在使用saver.restore恢復權值的時候,參數的設置需要萬分注意:


 
  1. # 先加載圖文件
  2. saver = tf.train.import_meta_graph("./model/save_model.meta")
  3. graph = tf.get_default_graph()
  4. # 其中的一個節點
  5. a_val = graph.get_tensor_by_name("var/a_val:0") # 這里是我的節點名稱,這都不是重點
  6.  
  7. with tf.Session() as sess:
  8. # 加載權值
  9. saver.restore(sess, "./model/save_model") # 這里才是重點,注意使用的是save_model,而不是save_model.data-00000-of-00001,不要帶一串后綴,如果使用后者的話,會報錯?。。。?!
  10. print(sess.run(a_val))

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

分享到:

相關信息

系統教程欄目

欄目熱門教程

人氣教程排行

站長推薦

熱門系統下載