Tensorflow关卡模式越来越删除

Question

我使用tensorflow检查点使用下面的代码，每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所需的值。