Tensorflow import_meta_graph返回'张量不存在'错误

时间:2017-02-06 15:35:12

标签: tensorflow

我正在尝试在Tensorflow中导入已保存的神经网络。我在培训后保存了它:

saver = tf.train.Saver()
saver.save(sess, filename)

在我用于推理的脚本中,我将其恢复为:

sess = tf.Session()
saver = tf.train.import_meta_graph(filename.meta)
saver.restore(sess, tf.train.latest_checkpoint('./'))

但在import_meta_graph行期间,我收到此错误:

  

KeyError:"名称' dropout1 / cond / dropout / Shape / Switch:1'是指不存在的张量。图表中不存在操作' dropout1 / cond / dropout / Shape / Switch'。"

我查看了原始笔记本中的张量和操作的名称,我在其中训练了模型,并且确实存在错误消息中提到的名称。此外,我使用相同的代码来保存和导入其他模型,它的工作原理。唯一的区别是我在AWS机器上训练了这些,使用旧版本的tensorflow,而我在计算机上训练了有问题的那个。

0 个答案:

没有答案