每个时期后如何保存我的keras模型?

时间:2020-01-08 18:21:26

标签: tensorflow keras model save epoch

我有正在使用的工作图像分割模型。我一直在调整model.fit()的参数以获得最佳结果。目前,它的运行时间约为1000 epochs。我想知道是否有一种方法可以在每个model.save()之后添加nth,而不是仅在最后一个周期之后保存模型?我想这样做,以便可以进行预测,以显示模型精度如何提高,并制作一个简短的动画来显示随时间变化的图像分割。

1 个答案:

答案 0 :(得分:0)

已解决:

ModelCheckpoint('models\\unet_seg_{epoch:04d}.model', monitor=observe_var, save_best_only=False, period = 2)