你如何在 Keras 中继续训练以前的模型？

Posted 2023-02-23

【中文标题】你如何在 Keras 中继续训练以前的模型？

【英文标题】：How do you keep training off a previous model in Keras?

【发布时间】：2020-06-16 04:54:08

【问题描述】：

假设我已经在 500 张图像上训练了一个模型并将其保存为 HDF5 文件。现在我有 90 多张图像要用于训练。我该如何做才能改进模型，而不是用 500 张图像覆盖模型？

【问题讨论】：

这能回答你的问题吗？ Loading a trained Keras model and continue training

【参考方案1】：

1) 由于数据集的大小非常小 (500)，您可以将所有数据一起重新训练。

其他

2) 您可以在 90 张图像上尝试以较小的学习率进行迁移学习。保持较小的学习率不会显着改变已经训练好的权重。

关于迁移学习的更多信息 https://towardsdatascience.com/transfer-learning-from-pre-trained-models-f2393f124751