Tensorflow。无法打开Checkpoint

时间:2016-08-16 18:28:14

标签: tensorflow

我已经构建并训练了张量流模型,但不幸的是,检查点文件无法打开,如下所示。

Now there isn't an error, but a bunch of warnings that don't really tell you anything.

当我运行评估代码时会发生这种情况:

import tensorflow as tf

import main
import Process
import Input

eval_dir = "/Users/Zanhuang/Desktop/NNP"
checkpoint_dir = "/Users/Zanhuang/Desktop/NNP"

ckpt = tf.train.get_checkpoint_state(checkpoint_dir)

def evaluate():
  with tf.Graph().as_default() as g:
    images, labels = Process.eval_inputs()
    forward_propgation_results = Process.forward_propagation(images)
    init_op = tf.initialize_all_variables()
    saver = tf.train.Saver()
    top_k_op = tf.nn.in_top_k(forward_propgation_results, labels, 1)

  with tf.Session(graph = g) as sess:
    tf.train.start_queue_runners(sess = sess)
    sess.run(init_op)
    saver.restore(sess, eval_dir)

    if ckpt and ckpt.model_checkpoint_path:
        saver.restore(sess, ckpt.model_checkpoint_path)

    for i in range(100):
        print(sess.run(top_k_op))

def main(argv = None):
    evaluate()

if __name__ == '__main__':
  tf.app.run()

接下来是我如何生成检查点文件:

                if step % 2 == 0:
                    checkpoint_path = os.path.join(FLAGS.data_dir, 'model.ckpt')
                    saver.save(sess, checkpoint_path, global_step = step)

0 个答案:

没有答案