-
TensorFlow实战框架(第二版)Chp5,P113 的Saver无法正常读取产生的Key v1_1 not found in checkpoint问题
-
解决方法:在 saver.restore()加载的前面,需要添加 tf.reset_default_graph(),作用是 清除默认图的堆栈,并设置全局图为默认图
# -*- coding: utf-8 -*-
"""
Created on Mon Oct 15 09:57:46 2018
@author: admin
"""
import tensorflow as tf
tf.reset_default_graph()
### 清除默认图的堆栈,并设置全局图为默认图
v1=tf.Variable(tf.constant(1.0,shape=[1]),name="v1")
v2=tf.Variable(tf.constant(2.0,shape=[1]),name="v2")
result=v1+v2
init_op=tf.global_variables_initializer()
saver=tf.train.Saver()
with tf.Session() as sess:
#sess.run(init_op)
saver.restore(sess,"../model/model.ckpt")
print(sess.run(result))