NotFoundError Key w1_1 not found in checkpoint

NotFoundError Key w1_1 not found in checkpoint

 admin  学习笔记, 数据分析  围观334次  2 条评论  编辑日期：2018-04-24  字体：大 中 小

一、描述：

在使用tf.train.Saver()类保存完训练好的模型参数后，在预测或者用到之前的参数时候，需要加载保存的参数，但是在第一次读取的时候没有问题，多次读取后出现参数name变化，并且在索引中找不到的bug,如下：

1	NotFoundError ( see above for traceback ) : Key w1_1 not found in checkpoint

二、原因

当前计算图使用的是默认的计算图，在第一次正常加载完后，已经有w1的变量，当再次加载时，因为name重复，自动将其改为w1_1，这样就与保存的模型变量参数不一致，出现error.

三、解决办法

1.将编译器的环境重置，会重新加载第一次的信息，但是每次需要重新加载时，都要重置，麻烦。

2.在程序的最后加上

1	tf . reset_default_graph ( ) #清除当前默认图中堆栈，重置默认图，实现模型参数的多次读取

清除当前默认图中堆栈，重置默认图，实现模型参数的多次读取

• 本文固定链接: http://www.bingbig.com/261.html