Я новичок в tenorflow. У меня есть модель, которая действительно означает и вычисление дисперсии для вектора, для которого передается размер. Я хочу сохранить модель и использовать ее для расчета для разных размеров.
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 как Tensor: имя
«save / Const: 0» относится к Tensor, который не существует. Операция,
'save / Const', не существует на графике.
Пожалуйста, дайте мне знать, как это исправить?
Спасибо