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

你可能感兴趣的:(Not found: Key Variable_1 not found in checkpoint)