我正在使用在tensorflow v2中定义为子模块的keras。我正在使用fit_generator()
方法训练模型。我想每10个时间保存一次模型。我该如何实现?
在Keras(不是tf的子模块)中,我可以给出ModelCheckpoint(model_savepath,period=10)
。但在TF V2,他们已经改变了这ModelCheckpoint(model_savepath, save_freq)
地方save_freq
可'epoch'
在这种情况下,模型保存每个时间段。如果save_freq
为整数,则在处理了许多样本后将保存模型。但我希望在10个时代之后。我该如何实现?
WARNING:tensorflow:'period' argument is deprecated. Please use 'save_freq' to specify the frequency in number of samples seen.
因此,我想此功能即将推出。在这种情况下,我该如何实现?