名称'save / Const:0'表示不存在的张量。图形中不存在“保存/常量”操作

时间:2019-07-03 20:42:48

标签: tensorflow neural-network

我是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'表示不存在的张量。操作,   图中不存在“保存/常量”。

请让我知道如何解决此问题? 谢谢

0 个答案:

没有答案