如何在 Keras 中显式调用 ModelCheckpoint(或任何回调)?



我的模型检查点旨在保存每个纪元的模型。不幸的是,我不使用模型的内置纪元。

当我的代码结束纪元时,如何不通过回调而是显式调用 ModelCheckpoint,以便它完成所有工作?

我找到了方法on_epoch_end,但不知道,在哪里传递模型本身?

每个回调都派生自Callback抽象基类。它具有set_model用于传递相应模型的方法。在 callbacks.py 中检查

相关内容

最新更新