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.在程序的最后加上

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

猜你喜欢

转载自blog.csdn.net/zzx3163967592/article/details/80780109
今日推荐