tf.train.get_checkpoint_state alwaysなし

Question

Tensorflow保存モデルを復元するための有効なチェックポイントファイルがあるかどうかを確認するには、tf.train.get_checkpoint_stateを使用しています。

ckpt = tf.train.get_checkpoint_state(os.path.dirname('fi/saves')) 
if ckpt and ckpt.model_checkpoint_path: 
    saver.restore(sess, ckpt.model_checkpoint_path) 

私の問題は、ckptは常に「なし」です。

• stats.json
• variables.ckpt.data-00000-の-00001
• variables.ckpt.index

internals.pkl：ここに私の内容はディレクトリ節約です

復元の呼び出しが正常に機能することに注意してください。

編集：私はこのようなlatest_filename引数を使用してみましたが

：私はまだなしを取得していないのです

tf.train.get_checkpoint_state(os.path.dirname(checkpoint_dir), latest_filename='variables.ckpt.index') 

。

Answer 1

私はtf.train.get_checkpoint_stateがチェックポイントファイルを探すことを知りました。

私はこれをやっていたので、保存するときに、私は、このファイルを持っていなかった。

saver.save(sess, variables_file_path, write_meta_graph=False, write_state=False) 

の代わりに：

saver.save(sess, variables_file_path, write_meta_graph=False, write_state=True) # write state is true by default. 

これを機能するようになりました！