Not found: Key Variable_1 not found in checkpoint

最近在写Tensorflow的时候遇到了这种问题,也确实让我困惑了很久。最后经过多方查找资料,发现原来是因为自己在创建参数的过程中,采用的是函数的形式,而的训练的过程中,保存下来的model里面的参数名与预测采用模型的参数名,不同导致~没错,就是保存的过程中偷偷地给你修改了。。。
我个人认为是因为采用的的tf.train.Saver(tf.global_variables()),保存的是全局变量。
最终解决方法:将需要的变量设定为全局,也就是从函数中移动到外边。

参考:
http://blog.csdn.net/l18930738887/article/details/69808692

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容