亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

在不顯式保存會話變量的情況下恢復TensorFlow模型

在不顯式保存會話變量的情況下恢復TensorFlow模型

江戶川亂折騰 2021-05-07 14:18:16
我看過許多有關保存訓練有素的神經網絡的問題,包括Tensorflow:如何保存/恢復模型?和https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/,但是在我沒有明確地將特定變量與模型一起保存的情況下,它們都不保存模型。這是我的情況:# In session "sesh" saver = tf.train.Saver() saver.save(sesh,os.getcwd(),latest_filename= 'RNN_plasma.ckpt')現在,我退出了該會話并想要恢復我剛剛保存的模型。我怎樣才能做到這一點?嘗試時:import tensorflow as tfwith tf.Session() as session1:    #First let's load meta graph and restore weights    saver = tf.train.import_meta_graph('RNN_plasma.ckpt')#error-line    saver.restore(session1,tf.train.latest_checkpoint('./')),則tf.train.import_meta_graph()調用返回:raise IOError("Cannot parse file %s: %s." % (filename, str(e)))IOError: Cannot parse file RNN_plasma.ckpt: 1:1 : Message type "tensorflow.MetaGraphDef" has no field named "model_checkpoint_path"..任何人都可以對這里發生的事情以及如何解決問題有任何見解嗎?(我的TensorFlow版本沒有tf.python.saved_model.simple_save()附帶(我有git_version 1.5.0))
查看完整描述

2 回答

?
HUWWW

TA貢獻1874條經驗 獲得超12個贊

保存:


saver = tf.train.Saver()

saver.save(sess,"/tmp/network")

恢復:


sess =  tf.Session() 

saver = tf.train.import_meta_graph('/tmp/network.meta')

saver.restore(sess,tf.train.latest_checkpoint('/tmp'))

graph = tf.get_default_graph()


查看完整回答
反對 回復 2021-05-25
  • 2 回答
  • 0 關注
  • 195 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號