无法加载预训练模型

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.

如何解决这个问题？

我已经提到这个post：

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

Answer 1

导入器在您的图表中找不到非常具体的函数，即 attn_add_fun_f32f32f32，这很可能是 one of attention functions。

可能您已经进入了this issue。但是，他们说它捆绑在 tensorflow 1.0 中。仔细检查已安装的 tensorflow 版本是否包含 attention_decoder_fn.py（或者，如果您使用的是另一个库，请检查它是否存在）。

如果有，您可以选择以下选项：

, if possible. You might want to read this discussion 解决方法。
复制您的图形定义，这样您就不必调用 import_meta_graph，而是将模型恢复到当前图形中。

无法加载预训练模型

Unable to load a pretrained model

python

machine-learning

tensorflow

neural-network

chatbot