我的模型检查点旨在保存每个纪元的模型。不幸的是,我不使用模型的内置纪元。
当我的代码结束纪元时,如何不通过回调而是显式调用 ModelCheckpoint,以便它完成所有工作?
我找到了方法on_epoch_end
,但不知道,在哪里传递模型本身?
每个回调都派生自Callback
抽象基类。它具有set_model
用于传递相应模型的方法。在 callbacks.py 中检查