天天看点

报错:Key_XXXXXX not found in checkpoint 的一种最基本的可能问题原因分析:

原因分析:

在其他博客里给出的解决办法都是针对确认保存没问题的情况下给出的,利用tf.reset_default_graph()清除堆栈什么的。但在跑自己改进后的网络时并不能解决以上报错。同桌提醒我再检查一下有没有动网络保存和恢复的部分。 这我就想起来了,之前为了修改的网络训练方便,把恢复设定成了部分网络恢复,这样可以在较好的模型的基础上继续训练自己添加的局部网络。但是我修改了tf.Saver(),这就导致保存的时候也是保存我指定的那部分网络,也就是说我保存的图不完整。如果是接着已有模型训练的话是可以读入部分网络继续训练的,但当进行测试时需要读入完整的模型才能运行,这时各种节点找不到的问题就出现了。 这也让我突然想起来,为什么我每次恢复模型继续训练的时候首轮结果总是和前一次训练结果差很多,原来是因为我没有恢复完整。这里感谢一下可能看不到我这篇文章的小同桌~