不显式地恢复TensorSession模型变量

2024-05-20 14:38:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我看过很多关于保存训练过的神经网络的问题,包括Tensorflow: how to save/restore a model?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 tf

with 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_图形()呼叫返回:

^{pr2}$

有谁能对这里发生的事情和如何解决它有任何见解吗?在

(我的TensorFlow版本没有tf.python.saved_模型.simple\u save()。(我有git_1.5.0版)


Tags: 模型importsavetftensorflowtrainrestorelatest
2条回答

保存:

saver = tf.train.Saver()
saver.save(sess,"/tmp/network")

还原:

^{pr2}$

您保存了一个简单的检查点,但随后尝试将其作为元图加载。这行不通。 在TensorFlow网站上有一篇文章解释了这些区别

https://www.tensorflow.org/mobile/prepare_models#what_is_up_with_all_the_different_saved_file_formats

必须有一个以.meta结尾的文件。在

相关问题 更多 >