【发布时间】:2017-07-06 15:44:26
【问题描述】:
我的ModelCheckpoint 旨在在每个时期保存模型。不幸的是,我不使用Model 的内置时期。
当我的代码结束 epoch 时,如何不通过回调而是显式调用 ModelCheckpoint,以便它完成所有工作?
我找到了方法on_epoch_end,但不知道在哪里传递模型本身?
【问题讨论】:
标签: machine-learning callback keras
我的ModelCheckpoint 旨在在每个时期保存模型。不幸的是,我不使用Model 的内置时期。
当我的代码结束 epoch 时,如何不通过回调而是显式调用 ModelCheckpoint,以便它完成所有工作?
我找到了方法on_epoch_end,但不知道在哪里传递模型本身?
【问题讨论】:
标签: machine-learning callback keras
每个回调都派生自 Callback 抽象基类。它具有set_model 方法,用于传递相应的模型。签到callbacks.py
【讨论】: