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

Save model every 10 epochs tensorflow.keras v2(每10个时期保存一次模型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的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:每10个时期保存一次模型tensorflow.keras v2

基础教程推荐