来源:https://tf.wiki/zh_hans/deployment/export.html
Checkpoint 只保存模型的参数,不保存模型的计算过程,因此一般用于在具有模型源代码的时候恢复之前训练好的模型参数。
而作为模型导出格式的 SavedModel 则更进一步,其包含了一个 TensorFlow 程序的完整信息:不仅包含参数的权值,**还包含计算的流程(即计算图)。当模型导出为 SavedModel 文件时,无须模型的源代码即可再次运行模型,这使得 SavedModel 尤其适用于模型的分享和部署。**后文的 TensorFlow Serving(服务器端部署模型)、TensorFlow Lite(移动端部署模型)以及 TensorFlow.js 都会用到这一格式。
本文对比了Checkpoint与SavedModel两种模型保存方式。Checkpoint仅保留模型参数,适合在有源代码情况下恢复模型。而SavedModel则包含了完整的TensorFlow程序信息,包括参数与计算流程,无需源代码即可运行,适用于模型分享和部署。
1535

被折叠的 条评论
为什么被折叠?



