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