浅谈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
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
CentOS7.4手动安装MySQL5.7的方法
PHP面向对象教程之自定义类
在 WordPress 的页眉(header)和页脚(footer)添加代码方法
![](https://img.aspxhome.com/file/2023/5/132435_0s.jpg)
Python中shapefile转换geojson的示例
一文搞懂Golang文件操作增删改查功能(基础篇)
![](https://img.aspxhome.com/file/2023/7/134297_0s.jpg)
Django实现自定义404,500页面教程
Python对多个sheet表进行整合实例讲解
Tensorflow分类器项目自定义数据读入的实现
![](https://img.aspxhome.com/file/2023/4/59144_0s.png)
微信小程序实现搜索框功能
![](https://img.aspxhome.com/file/2023/4/136904_0s.jpg)
python 制作网站小说下载器
![](https://img.aspxhome.com/file/2023/4/64984_0s.png)
python time模块用法实例详解
浅析vue中的nextTick
![](https://img.aspxhome.com/file/2023/9/133119_0s.png)
python3将视频流保存为本地视频文件
![](https://img.aspxhome.com/file/2023/1/89281_0s.jpg)
js仿支付宝多方框输入支付密码效果
![](https://img.aspxhome.com/file/2023/0/135630_0s.gif)
python中os和sys模块的区别与常用方法总结
numpy.ndarray 交换多维数组(矩阵)的行/列方法
Go高级特性探究之对象比较详解
也谈access数据库的防下载保护
原生js实现简易计算器
![](https://img.aspxhome.com/file/2023/5/126885_0s.jpg)