首页 > 学习笔记 > NotFoundError Key w1_1 not found in checkpoint
2018
04-24

NotFoundError Key w1_1 not found in checkpoint

一、描述:

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

二、原因

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

三、解决办法

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

2.在程序的最后加上

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

最后编辑:
作者:admin
这个作者貌似有点懒,什么都没有留下。

NotFoundError Key w1_1 not found in checkpoint》有 2 条评论

  1. 小仙女 说:

    棒棒

留下一个回复

你的email不会被公开。