第一卷 第十八章 Checkpointing模型
在第13章中,我们讨论了如何在训练完成后将模型保存和序列化到磁盘。在上一章中,我们学习了如何在发生时发现欠拟合和过拟合,从而使您能够终止表现不佳的实验,同时保持模型在训练时表现出的希望。
但是,您可能想知道是否可以将这两种策略结合起来。每当我们的损失/准确性提高时,我们可以序列化模型吗?或者是否可以仅序列化训练过程中的最佳模型(即损失最低或准确率最高的模型)?你打赌。幸运的是,我们也不必构建自定义回调——此功能已直接集成到Keras中。
1、检查点神经网络模型改进
检查点的一个很好的应用是在训练期间每次有改进时将您的网络序列化到磁盘。我们将“改进”定义为损失的减少或准确性的提高——我们将在实际的Keras回调中设置此参数。