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

问题描述 投票:0回答:1

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

tensorflow keras model save epoch
1个回答
0
投票

已解决:

ModelCheckpoint('models\\unet_seg_{epoch:04d}.model', monitor=observe_var, save_best_only=False, period = 2)
© www.soinside.com 2019 - 2024. All rights reserved.