错误:张量图与会话图不同

Ume*_*mer 6 machine-learning conv-neural-network deep-dream tensorflow

我正在尝试从检查点文件加载以前训练过的张量训练模型,现在这些检查点文件中有操作变量,因此要加载图形,我必须首先从 **ckpt.meta 文件加载 graph_def:

graph = tf.Graph()
sess = tf.InteractiveSession(graph=graph)
saver = tf.train.import_meta_graph('/data/model_cache/model.ckpt-39.meta')
    ckpt = tf.train.get_checkpoint_state(FLAGS.checkpoint_dir)
    if ckpt and ckpt.model_checkpoint_path:
       if os.path.isabs(ckpt.model_checkpoint_path):
          saver.restore(sess, ckpt.model_checkpoint_path)
Run Code Online (Sandbox Code Playgroud)

加载模型后,我有一种方法可以使用此模型进行推理以实现深梦算法。问题是当我使用默认会话调用 eval 时,出现以下错误:

File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 555, in eval
return _eval_using_default_session(self, feed_dict, self.graph, session)File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework   /ops.py", line 3495, in _eval_using_default_session
raise ValueError("Cannot use the given session to evaluate tensor: "
ValueError: Cannot use the given session to evaluate tensor: the tensor's graph is different from the session's graph.
Run Code Online (Sandbox Code Playgroud)

我已经确认 tf.get_default_graph() 和 sess.graph 指向相同的内存地址。必须有一些非常基本的东西我错过了。

我是 tensorflow 的新手,因此在这方面的任何帮助将不胜感激。谢谢

Jon*_*iev 0

您正在导入的元图(即/data/model_cache/model.ckpt-39.meta)很可能与检查点的元图不同tf.train.get_checkpoint_state(FLAGS.checkpoint_dir)与检查点使用的

通常的做法是调用get_checkpoint_state()(或tf.train.latest_checkpoint(FLAGS.checkpoint_dir))并在调用中使用它的输出import_meta_graph(),然后使用相同的检查点名称(和返回的保护程序)恢复会话中的变量。当然,如果元图保存在每个检查点中,则可以完成此操作。