tensorflow - Tensorflow チェックポイント モデルが削除される

Question

次のコードを使用して、10 エポックごとにテンソルフロー チェックポイントを使用しています。

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))

問題は、新しいファイルが生成されると、以前の 5 つのモデル ファイルが自動的に削除されることです。

Answer 1

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