Tensorflow 停止并恢复训练

Question

我正在使用 Tensorflow 来训练我的模型。我通常每 10 个时期保存一次我的模型。我要训练的样本数量有限，因此我正在扩充我的数据集以制作更大的训练数据集。

如果我需要在断电后使用我保存的模型恢复训练，最好使用相同的数据集恢复训练还是创建一个新的数据集？

Answer 1

您的问题很大程度上取决于您如何扩充数据集。如果您的增强扭曲了基础数据集的统计分布，那么您应该使用断电前数据集恢复训练。否则，您就假设您的增强没有改变数据集的分布。

做出一个相当安全的假设（假设您的增强不会以极其显着的方式改变数据）您可以安全地在新数据集或旧数据集上恢复训练，而不会显着改变准确性。