在Jupyter笔记本中中断keras培训是否可以保存培训?

时间:2018-11-15 04:54:53

标签: keras jupyter-notebook

因此,我当时正在使用Jupyter Notebook和Keras进行机器学习项目,然后开始培训。几个小时后,我回来了,才意识到我不小心将历元设置得很高。

我想知道,如果我停止运行单元(发送KeyboardInterrupt),整个培训会被取消吗?还是仍然保存我当前所在时代的权重?

我仍然可以从下一个单元格访问模型。

2 个答案:

答案 0 :(得分:1)

经过训练的模型仍将保留在内存中,处于KeyboardInterrupt发生时的状态。只要不停止Python内核或不重新建立模型,您就可以继续使用经过训练的模型。要对此进行测试,请评估模型的预测准确性。

请注意,如果您继续训练模型,则KeyboardInterrupt会重新启动时期计数器。这将影响所有依赖时期号的回调。

答案 1 :(得分:0)

如果您尚未定义ModelCheckpoint回调或某些自定义模型保护程序回调,那么答案是否定的。

下次您应该添加ModelCheckpoint回调,这样在每个时期您的模型都会被保存,您可以恢复它