2017-03-22 18 views
0

Tensorflowをr1.0にアップグレードすると、restoreコマンドが機能していないように見えます。 たとえば、誰が次のことが間違っているのか教えてください。最後の行から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

関連する問題