Tensorflow加载SavedModel模型过程源码阅读

最新推荐文章于 2025-09-11 05:45:36 发布

原创

最新推荐文章于 2025-09-11 05:45:36 发布 · 754 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

首先关于SavedModel格式的介绍见Tensorflow SavedModel。
Tensorflow源代码中关于SavedModel加载的函数定义文件位置如下图所示。
在这里插入图片描述
最顶层的函数叫做LoadSavedModel，其中调用了LoadSavedModelInternal。

进入LoadSavedModelInternal，其中加载参数的主要操作调用了RestoreSession。

进入RestoreSession，可见到函数中主要进行了两个操作，一个是查找restore_op，并使用session执行restore_op来恢复参数，另一个是查找saved_model里面定义的init op（通常是NoOp）并使用session执行init op，我们主要关注restore_op。
在这里插入图片描述
查找restore_op，得到位置如下图所示。