首先关于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_
Tensorflow加载SavedModel模型过程源码阅读
最新推荐文章于 2022-02-15 17:50:10 发布