我是Tensorflow的新手。我有一个模型,该模型在传递大小的向量上进行均值和方差计算。我想要保存模型并将其用于不同大小的计算。
import tensorflow as tf
def mean_var(x):
vec = tf.Variable(tf.random_normal([x]))
mean, variance = tf.nn.moments(vec, [0], keep_dims=True)
return vec, mean, variance
with tf.Graph().as_default():
x = tf.constant(4, name='x')
vec, mean, variance = mean_var(x)
init = tf.initialize_all_variables()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
save_path = saver.save(sess, "mean_var.ckpt")
print("Model saved in file: %s" % save_path)
tf.reset_default_graph()
with tf.Session() as sess:
v1 = tf.constant(5, name='v1')
saver.restore(sess, "mean_var.ckpt")
print(v1.eval())
为此,我收到以下错误。
TypeError:无法将feed_dict键解释为张量:名称 'save / Const:0'表示不存在的张量。操作, 图中不存在“保存/常量”。
请让我知道如何解决此问题? 谢谢