python - 在 Keras 中保存模型检查点与保存整个模型

Question

哪种方法最好，无论是保存模型检查点还是每个时期都将整个模型保存到磁盘。为什么没有人保存整个模型？

标签： pythontensorflowkerasdeep-learning

Answer 1

检查点用于保存您的模型，以防万一您的系统崩溃或训练时代码中断，因此当您在崩溃后再次开始训练模型时，您不必从头开始。检查点捕获所有参数的确切值（tf.Variable对象）由模型使用。检查点不包含模型定义的计算的任何描述。

另一方面，SavedModel 格式除了参数值（检查点）之外，还包括模型定义的计算的序列化描述。这种格式的模型独立于创建模型的源代码。您可以在tensorflow的官方文档中看到上述信息。@R南塔克