2017-02-05 5 views
4

ために、一致するファイルが見つかりませんでしたが、tensorflowは、一致するファイルの場所を見つけることができないようです: - モデルを保存する失敗TensorSliceReaderコンストラクタに:私は私のモデルを保存しようとしたし、それを復元しようとした

コード出力: -

import tensorflow as tf 

save_file = 'model.ckpt' 


weights = tf.Variable(tf.truncated_normal([2, 3])) 
bias = tf.Variable(tf.truncated_normal([3])) 


saver = tf.train.Saver() 


with tf.Session() as sess: 

    sess.run(tf.global_variables_initializer()) 
    saver.save(sess, save_file) 

コードを復元するためのモデル

import tensorflow as tf 

save_file = 'model.ckpt' 
tf.reset_default_graph() 
weights = tf.Variable(tf.truncated_normal([2, 3])) 
bias = tf.Variable(tf.truncated_normal([3])) 
saver = tf.train.Saver() 

with tf.Session() as sess: 
    saver.restore(sess, 'model.ckpt') 

は、私は、エラーの下に取得しています: -

]が見つかりません:tensorflow /コア/フレームワーク/ op_kernel.ccは10

W失敗TensorSliceReaderコンストラクタ:975]:model.ckpt

Wのtensorflow /コア/フレームワーク/ op_kernel.ccため、一致するファイルが見つかりませんでした見つからない:失敗したTensorSliceReaderコンストラクタ:model.ckptに一致するファイルが見つかりません

+1

''。/ model.ckpt''を 'saver.restore()'に渡すとうまくいくのですか? – mrry

+0

うまくいきました!私はテンソルフローがパスを提供することを期待しているが、保存中に現在のディレクトリに保存されると思います。 – Schumi08

+0

はい...それはかなり奇妙な不一致です、私は恐れています。 [この問題](https://github.com/tensorflow/tensorflow/issues/6893)は問題を追跡していますが、TensorFlow 1.0以降はまだ開いています。 – mrry

答えて

4

saver.restore()メソッドは、ファイル名ではなくパスを第2引数として渡さない限り失敗します。この問題を回避するには、チェックポイントを含むディレクトリからスクリプトを実行している場合は、saver.restore(sess, './model.ckpt')を呼び出すことができます。

関連する問題