Tensorflow 停止并恢复训练

Question

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

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

Answer 1

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

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