不成功的TensorSliceReader构造函数：无法找到任何匹配的文件

Question

我尝试保存我的模型然后尝试恢复它，但似乎tensorflow无法找到匹配文件的位置： -

保存模型输出的代码： -

import tensorflow as tf

save_file = 'model.ckpt'


weights = tf.Variable(tf.truncated_normal([2, 3]))
bias = tf.Variable(tf.truncated_normal([3]))


saver = tf.train.Saver()


with tf.Session() as sess:

    sess.run(tf.global_variables_initializer())
    saver.save(sess, save_file)

恢复模型的代码

import tensorflow as tf

save_file = 'model.ckpt'
tf.reset_default_graph()
weights = tf.Variable(tf.truncated_normal([2, 3]))
bias = tf.Variable(tf.truncated_normal([3]))
saver = tf.train.Saver()

with tf.Session() as sess:
    saver.restore(sess, 'model.ckpt')

我收到以下错误： -

W tensorflow / core / framework / op_kernel.cc：975]未找到：不成功的TensorSliceReader构造函数：无法找到model.ckpt的任何匹配文件

W tensorflow / core / framework / op_kernel.cc：975]未找到：不成功的TensorSliceReader构造函数：无法找到model.ckpt的任何匹配文件

Answer 1

saver.restore()方法将失败，除非您传递路径 - 而不仅仅是文件名 - 作为第二个参数。要解决此问题，如果从包含检查点的目录运行脚本，则可以调用saver.restore(sess, './model.ckpt')。