python-3.x - 在 VGG-16 上停止和重新开始训练

Question

我正在使用预训练的 VGG-16 模型进行图像分类。我正在添加自定义最后一层，因为我的分类类的数量是 10。我正在为模型训练 200 个时期。

我的问题是：如果我在某个时期随机停止（通过关闭 python 窗口）训练，有什么办法吗？比如说没有。50 和从那里恢复？我已经阅读了有关保存和重新加载模型的信息，但我的理解是这仅适用于我们的自定义模型，而不适用于 VGG-16 等预训练模型。

标签： python-3.xmachine-learningkerascheckpointingvgg-net

Answer 1

您可以使用ModelCheckpoint回调定期保存模型。要使用它，请将callbacks参数传递给该fit方法：

from keras.callbacks import ModelCheckpoint
checkpointer = ModelCheckpoint(filepath='model-{epoch:02d}.hdf5', ...)
model.fit(..., callbacks=[checkpointer])

然后，稍后您可以加载最后保存的模型。有关此回调的更多自定义，请查看文档。