在使用tensorflow训练数据并使用把训练结果储存至 ‘/*.ckpt’文件以后,调用该文件以新数据集测试的时候,会出现报错。
本文基于jupyter notebook环境,pycharm的话还没试过大家可以试一下,欢迎讨论。
Key Variable_10 not found in checkpoint
当初查看源码语法感觉百思不得其解,后来查阅网上发现问题要解决其实挺简单,解决的办法有很多,个人觉得以下两个办法较为实在:
1、重新启动kernal
注意,只加载和测试相关的代码(几个小时辛苦训练的结果已经保存了/(ㄒoㄒ)/~~,开始不知道怎么回事还踩了坑)。这样大概率能够正常加载。补充,如果测试的时候数据多于一个,第二个数据进入的时候同样的报错还会发生。这时需要用到第二个方法。
2、tf.reset_default_graph()
在循环语句中一上面这句为首句,可以轻松解决。
办法不唯一,欢迎补充。
想知道具体原理的小伙伴可以参考
link.