我正在使用预先训练的VGG-16模型进行图像分类。我要添加自定义的最后一层，因为我的分类类数是10。我正在训练200个时代的模型。
我的问题是：如果我在某个时期随机停止（通过关闭python窗口）培训，有什么办法，比如说时期否。 50，然后从那里继续？我已经阅读了有关保存和重新加载模型的信息，但我的理解是，该模型仅适用于我们的自定义模型，而不适用于像VGG-16这样的预训练模型。
最佳答案
您可以使用ModelCheckpoint回调定期保存模型。要使用它，请将callbacks参数传递给fit方法：
from keras.callbacks import ModelCheckpoint
checkpointer = ModelCheckpoint(filepath='model-{epoch:02d}.hdf5', ...)
model.fit(..., callbacks=[checkpointer])
然后，稍后您可以加载最后保存的模型。有关此回调的更多自定义信息，请参阅文档。