如何在 tensorflow 中持续训练新数据

Question

我用的是TF-slim训练花数据集，脚本是这样的。花卉数据集只有 5 个类。如果我给玫瑰添加了一些新的图像数据，或者添加了一个新的分类，火车1000步后我该怎么办？我是否需要删除已经训练好的数据，例如检查点文件？

Answer 1

Data Science Stack Exchange上存在类似的问题，其答案考虑了您的场景：

一旦模型经过训练并且获得可用于训练的新数据，您就可以加载以前的模型并对其进行训练。例如，您可以将模型保存为 .pickle 文件并加载它，并在新数据可用时对其进行进一步训练。请注意，为了使模型正确预测，新的训练数据应具有与过去数据相似的分布

我在自己的项目中也做了同样的事情，我从一个小数据集开始，随着时间的推移变得越来越大。添加新数据后，我从最后一个检查点重新训练模型。