显式张量流会话在Tensorflow / nmt中给出了提取错误

时间:2017-12-21 05:19:12

标签: tensorflow opennmt

这种tf.session工作正常:

with tf.Session(graph=self.infer_model.graph, config=utils.get_config_proto()) as sess:
          loaded_infer_model = model_helper.load_model(self.infer_model.model, self.ckpt, sess, "infer")

但我必须保持持久会话以便重用。因此,不要使用"来创建tf.session。声明,我创建了一个:

sess = tf.Session(
            graph=infer_model.graph, config=utils.get_config_proto())
loaded_infer_model = model_helper.load_model(
              infer_model.model, ckpt, sess, "infer")

但这会产生以下错误(在model_helper.load_model中):有人可以建议如何加载可以重复使用的显式会话吗?

  

文件" /home/pksingh/sans/app/nmt/model_helper.py" ;,第444行,在   load_model session.run(tf.tables_initializer())文件   " /usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py" ;,   第889行,运行run_metadata_ptr)文件   " /usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py" ;,   第1103行,在_run self._graph中,取,feed_dict_tensor,   feed_handles = feed_handles)文件   " /usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py" ;,   第414行,在init self._fetch_mapper = _FetchMapper.for_fetch(fetches)   文件   " /usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py" ;,   第242行,在for_fetch中返回_ElementFetchMapper(取,   contraction_fn)文件   " /usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py" ;,   第278行,在init' Tensor中。 (%S)' %(fetch,str(e)))ValueError:Fetch   争论不可能   被解释为张量。 (操作名称:" init_all_tables"操作:" NoOp"   不是此图的元素。)

1 个答案:

答案 0 :(得分:1)

实现相同目标的最佳选择是使用交互式会话。您可以初始化这样的交互式会话:

sess = tf.InteractiveSession()

访问此link了解详情。