每10个时期保存一次模型tensorflow.keras v2

问题描述

我使用的是TensorFlow v2中定义为子模块的keras。我正在使用fit_generator()方法训练我的模型。我想每10个纪元保存一次我的模型。我如何实现这一点？

在Kera中(不作为Tf的子模)，我可以给出ModelCheckpoint(model_savepath,period=10)。但在TFv2中，他们将其更改为ModelCheckpoint(model_savepath, save_freq)，其中save_freq可以'epoch'，在这种情况下，每个时期都会保存模型。如果save_freq为整数，则在处理了如此多的样本后保存模型。但我希望它是在10个纪元之后。我如何实现这一点？

---

解决方案

使用tf.keras.callbacks.ModelCheckpoint使用save_freq='epoch'并传递额外参数period=10。

虽然official docs中没有记录这一点，但这是一种方法(请注意，它记录了您可以通过period，只是没有解释它的作用)。