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

Save model every 10 epochs tensorflow.keras v2(每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，只是没有解释它的作用)。

这篇关于每10个时期保存一次模型tensorflow.keras v2的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！