我正在尝试在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,而我在计算机上训练了有问题的那个。