现在位置: 首页 > 保存模型
2020年02月18日 编程语言 ⁄ 共 3937字 评论关闭

Tensorflow 使用pb文件保存(恢复)模型计算图和参数实例详解 一.保存: graph_util.convert_variables_to_constants 可以把当前session的计算图串行化成一个字节流(二进制),这个函数包含三个参数:参数1:当前活动的session,它含有各变量 参数2:GraphDef 对象,它描述了计算网络 参数3:Graph图中需要输出的节点的名称的列表 返回值:精简版的GraphDef 对象,包含了原始输入GraphDef和session的网络和变量信息,它的成员函数SerializeToString()可以把这些信息串行

阅读全文