python - Tensorflow "Attempting to use uninitialized value ..."恢复时出错

Question

我做了一个RNN模型。训练结束后，我将其保存在 tf.Session() 中，如下所示。

#main.py
...(training)
saver = tf.train.Saver()
save_path = saver.save(sess, "Save data/RNN-model.ckpt")

在“run.py”中我尝试恢复保存的数据。

#run.py
...
with tf.Session() as sess:
    saver = tf.train.Saver()
    ckpt = tf.train.get_checkpoint_state('Save data/')
    if ckpt and tf.train.checkpoint_exists(ckpt.model_checkpoint_path):
        saver.restore(sess, ckpt.model_checkpoint_path)
    ...(training or testing again)

运行时，出现错误消息。

FailedPreconditionError : Attempting to use uninitialized value accuracy/total

但是，当我按照错误消息所述删除“run.py”中与准确性相关的代码时，我认为它似乎运行良好。

我错过了什么吗？任何评论或答案都会对我有帮助。

谢谢。

Answer 1

accuracy 操作包含一些不属于图形的局部变量，因此应手动初始化。在restore之后添加sess.run(tf.local_variables_initializer())将初始化局部变量。