为了账号安全,请及时绑定邮箱和手机立即绑定

在不显式保存会话变量的情况下恢复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 关注
  • 167 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号