无法加载预训练的模型

Question

在对我的模型进行了将近2天的训练之后，生成了3个文件：

best_model.ckpt.data-00000-of-00001
best_model.ckpt.index
best_model.ckpt.meta

其中， best_model是我的模型名称。 当我尝试使用以下命令导入模型时

with tf.Session() as sess:
  saver = tf.train.import_meta_graph('best_model.ckpt.meta')
  saver.restore(sess, "best_model.ckpt")

我收到以下错误

Traceback (most recent call last):

File "<stdin>", line 2, in <module>
File "/home/shreyash/.local/lib/python2.7/site-

packages/tensorflow/python/training/saver.py", line 1577, in 
import_meta_graph
    **kwargs)
  File "/home/shreyash/.local/lib/python2.7/site-

packages/tensorflow/python/framework/meta_graph.py", line 498, in import_scoped_meta_graph
    producer_op_list=producer_op_list)

File "/home/shreyash/.local/lib/python2.7/site-packages/tensorflow/python/framework/importer.py", line 259, in import_graph_def
    raise ValueError('No op named %s in defined operations.' % node.op)

ValueError: No op named attn_add_fun_f32f32f32 in defined operations.

如何解决这个问题？

我已经提到了这篇文章： TensorFlow，为什么保存模型后会有3个文件？

• 使用pip安装Tensorflow版本1.0.0
• Linux版本16.04
• python 2.7

Answer 1

导入程序无法在您的图形中找到非常具体的功能，即attn_add_fun_f32f32f32 ，它很可能是关注功能之一 。

可能您已涉足此问题 。 但是，他们说它捆绑在tensorflow 1.0中。 所安装tensorflow版本仔细检查包含attention_decoder_fn.py （或者，如果你使用的是其他库，检查它的存在）。

如果有，这里是您的选择：

• 如果可能，请重命名此操作 。 您可能需要阅读此讨论以了解解决方法。
• 复制您的图定义，这样您就不必调用import_meta_graph ，而是将模型还原到当前图中。