及时保存特定的TensorFlow检查点

Question

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

一点背景：

我正在创建一个强化学习模型，我希望在整个培训过程中保存最好的模型。为了做到这一点，我保持最高分，每当更新时保存一个检查点。

不幸的是，我的best_score检查点被删除了。据我所知，原因是TF只保留了最新的5个检查点，这很好。

我想要保留最近的5个检查点和最近的检查点，这可能不是最近的五个检查点。有没有办法在不存储所有检查点的情况下这样做？

谢谢大家！

Answer 1

查看已发布的问题here和here，这似乎是尚未实施的请求功能。您可以使用.babelrc阻止删除所有检查点。如果您正在做一些大事，那么为了避免填满您的磁盘，我建议您不要开始保存检查点，直到经过合理的步骤数，并且除非当前结果超过上次保存，否则不保存得到一些最低金额。