我的ModelCheckpoint旨在每轮(epoch)保存模型。不幸的是,我没有使用Model的内置轮次。
如何在代码结束一轮时,不通过回调而是显式地调用ModelCheckpoint,使其完成所有工作?
我找到了on_epoch_end
方法,但不知道如何传递模型本身?
回答:
每个回调都派生自Callback
抽象基类。它有一个set_model
方法,用于传递相应的模型。请查看callbacks.py
我的ModelCheckpoint旨在每轮(epoch)保存模型。不幸的是,我没有使用Model的内置轮次。
如何在代码结束一轮时,不通过回调而是显式地调用ModelCheckpoint,使其完成所有工作?
我找到了on_epoch_end
方法,但不知道如何传递模型本身?
回答:
每个回调都派生自Callback
抽象基类。它有一个set_model
方法,用于传递相应的模型。请查看callbacks.py