2016-12-07 11 views
2

を削除したばかり、私は次のコードを使用して、すべての10のエポック後tensorflowチェックポイントを使用しています:問題は、新しいファイルが生成され得ているとして、前回の5モデルファイルは自動的に削除得ている、ということであるTensorflowチェックポイントモデルは

checkpoint_dir = os.path.abspath(os.path.join(out_dir, "checkpoints")) 
checkpoint_prefix = os.path.join(checkpoint_dir, "model") 
... 
if current_step % checkpoint_every == 0: 
    path = saver.save(sess, checkpoint_prefix, global_step=current_step) 
    print("Saved model checkpoint to {}\n".format(path)) 

答えて

6

これは予想される動作です。tf.train.Saverのドキュメントでは、デフォルトで5つの最新のチェックポイントファイルが保持されているとしています。これを調整するには、max_to_keepを目的の値に設定します。

+0

ありがとう@Gregory。私はその部分を逃した。 – Nitin