我正在使用预训练的 VGG-16 模型进行图像分类。我添加了自定义最后一层，因为我的分类类数量为 10。我正在对模型进行 200 轮训练。
我的问题是：如果我随机停止（通过关闭 python 窗口）在某个时期的训练，有什么办法吗？假设时期没有。 50 然后从那里恢复？我读过有关保存和重新加载模型的内容，但我的理解是，这只适用于我们的自定义模型，而不适用于 VGG-16 等预训练模型。
您可以使用ModelCheckpoint回调以定期保存您的模型。要使用它，请传递一个callbacks参数到fit method:
from keras.callbacks import ModelCheckpoint
checkpointer = ModelCheckpoint(filepath='model-{epoch:02d}.hdf5', ...)
model.fit(..., callbacks=[checkpointer])
然后，稍后您可以加载上次保存的模型。有关此回调的更多自定义，请查看文档。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
