如何通过使用tf.keras.callbacks.ModelCheckpoint的训练步骤来保存模型

时间:2020-02-28 02:00:45

标签: python tensorflow keras nlp artificial-intelligence

我的训练数据非常大,这意味着每个时期都有许多训练步骤。我想使用tf.keras.callbacks.ModelCheckpoint每隔100,000个步骤保存一个模型,并将所有这些模型保存到磁盘中,而不是在保存新模型时重写模型文件。但是,我在阅读文档后不知道如何执行此操作。似乎此回调旨在按时代保存模型。

那么,我是否必须实现自定义回调以逐步保存模型?

0 个答案:

没有答案
相关问题