machine-learning - 及时保存特定的 TensorFlow 检查点

Question

是否可以标记不删除的检查点？

一些背景:

我正在创建一个强化学习模型，我想在整个训练过程中保存最好的模型。为了做到这一点，我会保持最好的分数，并在每次更新时及时保存检查点。

不幸的是，我的 best_score 检查点正在被删除。我明白原因是 TF 只保留最新的 5 个检查点，这很好。

我只想保留最近 5 个检查点以及可能不在最近 5 个检查点中的最佳检查点。有没有办法在不存储所有检查点的情况下做到这一点？

谢谢大家!

Answer 1

查看发布的问题 here和 here ，这似乎是一项请求的功能，但尚未实现。您可以使用 saver = tf.train.Saver(max_to_keep=0) 防止删除所有检查点。如果您正在做一些大的事情，那么为了防止它填满您的磁盘，我建议您在经过合理数量的步骤之前不要开始保存检查点，并且除非当前结果比上次保存的结果小一些，否则不要保存金额。