保存tensorflow计算图
import tensorflow as tf
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.save(sess,“F:/model/model.ckpt”)
加载tensorflow计算图
import tensorflow as tf
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
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess,“F:/model/model.ckpt”)
print(sess.run(result))
当执行这两段代码时,多次执行,会出现如下错误:
NotFoundError: Key v1_1 not found in checkpoint
在两段代码最后加上下列语句就可以了:
tf.reset_default_graph()