tensorflow - TensorFlow 检查点模型被删除

Question

我使用以下代码在每 10 个时期后使用 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))

问题是，随着新文件的生成，之前的 5 个模型文件会被自动删除。

Answer 1

这是预期的行为，tf.train.Saver的文档说默认情况下保留 5 个最近的检查点文件。要调整它，请将 max_to_keep 设置为所需的值。