tensorflow bug:NotFoundError Key w1_1 not found in checkpoint

bug:


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

多次保存模型参数

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

解决方法:保存模型最后加上

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

参考链接:http://www.bingbig.com/261.html

猜你喜欢

转载自blog.csdn.net/qq_40614981/article/details/81603930
今日推荐