小编Tia*_* Gu的帖子

我想使用tf.train.Saver()来制作张量的检查点,这是我的代码片段:

import tensorflow as tf

with tf.Graph().as_default():
    var = tf.Variable(tf.zeros([10]), name="biases")
    temp = tf.add(var, 0.1)
    init_op = tf.global_variables_initializer()

    saver = tf.train.Saver({'w':temp})

    with tf.Session() as sess:
        sess.run(init_op)
        print(sess.run(temp))

但得到如下错误:

Traceback (most recent call last):
  File "./test_counter.py", line 61, in <module>
    saver = tf.train.Saver({'w':temp})
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1043, in __init__
    self.build()
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1073, in build
    restore_sequentially=self._restore_sequentially)
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 649, in build
    saveables = self._ValidateAndSliceInputs(names_to_saveables)
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 578, in _ValidateAndSliceInputs
     variable)
   TypeError: names_to_saveables must be a dict mapping string …