标签： machine-learning callback keras
我的ModelCheckpoint旨在为每个纪元保存模型。不幸的是，我没有使用Model的内置时代。
如何通过回调调用ModelCheckpoint，但显然，当我的代码结束时代时，它会完成所有工作？
我找到方法on_epoch_end，但无法弄清楚，在哪里传递模型本身？
on_epoch_end
答案 0 :(得分：2)
每个回调都派生自Callback抽象基类。它有set_model方法，用于传递相应的模型。在callbacks.py
Callback
set_model