如何在每个纪元后保存我的 keras 模型?
How do I save my keras model after every epoch?
我有正在使用的工作图像分割模型。我一直在调整 model.fit()
的参数以获得最佳结果。目前 运行 大约 1000 epochs
。我想知道是否有办法在每个 nth
纪元之后包含 model.save()
而不是在最后一个纪元之后保存模型?我想这样做,以便我可以预测模型的准确性如何提高,并制作一个简短的动画来显示图像分割随时间的推移。
已解决:
ModelCheckpoint('models\unet_seg_{epoch:04d}.model', monitor=observe_var, save_best_only=False, period = 2)
我有正在使用的工作图像分割模型。我一直在调整 model.fit()
的参数以获得最佳结果。目前 运行 大约 1000 epochs
。我想知道是否有办法在每个 nth
纪元之后包含 model.save()
而不是在最后一个纪元之后保存模型?我想这样做,以便我可以预测模型的准确性如何提高,并制作一个简短的动画来显示图像分割随时间的推移。
已解决:
ModelCheckpoint('models\unet_seg_{epoch:04d}.model', monitor=observe_var, save_best_only=False, period = 2)