我怎样才能停止模型训练并恢复它？答案

【问题标题】：How can i stop model training and resume it?我怎样才能停止模型训练并恢复它？
【发布时间】：2021-08-05 08:21:58
【问题描述】：

我正在使用自主数据集进行对象检测。我想用 10000 个训练图像、2000 个测试、2000 个验证图像来训练我的模型。我将使用对象检测 tensorflow lite 模型制造商。

项目链接：tensorflow.org/lite/tutorials/model_maker_object_detection

设置批量大小 32 后，训练需要 50 个 epoch 并持续 2 天（第 3 步）。我无法让我的计算机两天开机。我正在 jupyter notebook 中运行项目

如何停止模型训练并再次恢复它？（例如停止第 10 个 epoch 并在一天后继续）

【问题讨论】：

【解决方案1】：

我确定这取决于您正在处理的代码。您可以使用 tensorflow 检查来做到这一点 How to Pause / Resume Training in Tensorflow

【讨论】：