小编fra*_*bon的帖子

我正在尝试在 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 机器上使用旧版本的张量流训练了这些，而我在我的计算机上训练了有问题的机器。