2017-03-22 196 views
0

将Tensorflow升级到r1.0后,恢复命令似乎不起作用。 例如,任何人都可以告诉我以下哪些问题?如何在Tensorflow中恢复变量r1.0

def foo(): 
    v1 = tf.Variable(1., name="v1") 
    v2 = tf.Variable(2., name="v2") 
    v3 = v1 + v2 

    saver = tf.train.Saver() 

    with tf.Session() as sess: 
     tf.global_variables_initializer().run() 

     saver.save(sess, "temp") 

     # do something 

     saver.restore(sess, "temp") 

从上线,我得到了一个错误:

tensorflow.python.framework.errors_impl.NotFoundError: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for temp 
    [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save/Const_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]] 

Tensorflow文档仍持有旧版本对此事的解释。

回答

1

TensorFlow 1.0有一个错误,它不能识别tf.Saver.restore()只包含文件名(并且没有路径组件)的文件名。这将在下一个版本中修复,但现在您应该可以使用以下解决方法添加路径组件:

saver.restore(sess, "./temp") 
+0

它的工作原理!非常感谢! – user3755060

相关问题