浅谈tf.train.Saver()与tf.train.import_meta_graph的要点
作者：.我心永恒_ 发布时间：2022-06-02 09:59:32
(一)、tf.train.Saver()
(1). tf.train.Saver() 是用来保存tensorflow训练模型的，默认保存全部参数
(2). 用来加载参数，注：只加载存储在data中的权重和偏置项等需要训练的参数，其他一律不加载，包括meta文件中的图也不加载
(二)、tf.train.import_meta_graph
(1). 用来加载meta文件中的图,以及图上定义的结点参数包括权重偏置项等需要训练的参数,也包括训练过程生成的中间参数,所有参数都是通过graph调用接口get_tensor_by_name(name="训练时的参数名称")来获取
(三)、总结
(1). 保存使用tf.train.Saver()
(2). 加载可以使用tf.train.import_meta_graph(".meta文件"),直接通过训练参数名称就可以获取需要的参数,但需要提前知道训练时的参数名称才能获取,要懂得tensorflow命名规则
(3). tf.train.Saver("./checkpoints目录/")加载的缺点是只加载了训练参数,并且必须定义与之相同(shape,dtype要相同,tf.type要相同，如：我是placeholder,你也是tf.placeholder)方能使用,当你要获取训练中间参数时,需要和训练过程一样搭建相同的网络.
补充：tf.train.import_meta_graph报KeyError
我在模型恢复时，在执行tf.train.import_meta_graph的时候报错
后来发现，我的模型是在服务器上训练的，服务器上tensorflow版本是1.11.0,而我在本地电脑上执行的 tf.train.import_meta_graph，我本地的tensorflow是1.5.0，我将tensorflow更新到1.11.0后，就解决了。
来源：https://blog.csdn.net/sinat_36618660/article/details/98665482
0
投稿猜你喜欢
- 最近一个项目中遇到ASP对FoxPro库表(*.DBF)的操作问题。现实中确有许多应用软件使
- NumPy是一个关于矩阵运算的库，熟悉Matlab的都应该清楚，这个库就是让python能够进行矩阵话的操作，而不用去写循环操作。下面对nu
- enumerate首先介绍的是enumerate函数。在我们日常编程的过程当中，经常会遇到一个问题。在C语言以及一些古老的语言当中是没有迭代
- 为新项目写的一份规范文档, 分享给大家. 我想前端开发过程中, 无论是团队开发, 还是单兵做站, 有一份开发文档做规范, 对开发工作都是很有
- 前言PyTorch是一个开源的深度学习框架，基础教学从开始安装学起，一步一个脚印。提示：以下是本篇文章正文内容，下面案例可供参考一、Pyor
1.过程蜘蛛纸牌大家玩过没有？之前的电脑上自带的游戏，用他来摸鱼过的举个手。但是现在的电脑上已经没有蜘蛛纸牌了。所以…- 一、概述OLAP的系统(即Online Aanalyse Process)一般用于系统决策使用。通常和数据仓库、数据分析、数据挖掘等概念联系
- 本文实例讲述了Python实现的列表排序、反转操作。分享给大家供大家参考，具体如下：排序：使用sorted方法和列表的sort方法：sort
- 本文实例讲述了python实现ip代理池功能。分享给大家供大家参考，具体如下：爬取的代理源为西刺代理。用xpath解析页面用telnet来验
- '去掉字符串头尾的连续的回车和空格 function trimVBcrlf(str) tr
- 错误信息: RuntimeError: in-place operations can be only used on variables
- 我们可用ADO STREAM来做一个无组件的上传程序。Stream对象包含了许多操作二进制和文本文件的方法，我们现在用Stream对象来操作
- 在实际生活中，经常会有文件重复的困扰，即同一个文件可能既在A目录中，又在B目录中，更可恶的是，即便是同一个文件，文件名可能还不一样。在文件较
- python字符串方法分类，字符串是经常可以看到的一个数据储存类型，我们要进行字符的数理，就需要用各种的方法，这里有许多方法，我给大家介绍比
- Pytorch中retain_graph的坑在查看SRGAN源码时有如下损失函数，其中设置了retain_graph=True，其作用就是在
- 假如Excel中的数据如下：数据库建表如下：其中Id为自增字段：代码：using System;using System.Collectio
- 目录什么是pyecharts?pyecharts安装加载折线图的绘制条形图和折线图的结合绘制漏斗图什么是pyecharts?pyechart
- 前言在我们往数据库插入数据的时候，需要判断某个字段是否存在，如果存在则执行更新操作，如果不存在则执行插入操作，如果每次首先查询一次判断是否存
- 通过界面设计上是能手工操作的，无法达到我批量修改几千台服务器。 因为此了一个脚本来批量执行。 环境：redgate + mssql 2008
- >>> a = 2.5>>> b = 2.5>>> c = b>>>
