2020
09-29
09-29
浅谈tensorflow模型保存为pb的各种姿势
一,直接保存pb1,首先我们当然可以直接在tensorflow训练中直接保存为pb为格式,保存pb的好处就是使用场景是实现创建模型与使用模型的解耦,使得创建模型与使用模型的解耦,使得前向推导inference代码统一。另外的好处就是保存为pb的时候,模型的变量会变成固定的,导致模型的大小会大大减小。这里稍稍解释下pb:是MetaGraph的protocolbuffer格式的文件,MetaGraph包括计算图,数据流,以及相关的变量和输入输出主要使用tf.SavedM...
继续阅读 >