浅谈tf.train.Saver()与tf.train.import_meta_graph的要点
(一)、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后,就解决了。
(资源库 www.zyku.net)
原文链接:https://blog.csdn.net/sinat_36618660/article/details/98665482
栏 目:Python教程
本文标题:浅谈tf.train.Saver()与tf.train.import_meta_graph的要点
本文地址:https://www.zyku.net/python/9768.html
您可能感兴趣的文章
- 07-05Linux minfo命令
- 09-24咪咕视频怎么取消自动扣款
- 10-27萤石云监控录像怎样删除
- 08-15Audition音频网格关闭步骤介绍
- 12-14小爱音箱如何重新连接手机
- 07-05Linux dip命令
- 12-23HTML <article> 标签
- 11-16iphone手机访问限制功能在哪里
- 01-08安康家政师傅-安康家政师傅应用软件功
- 04-05红魔6pro开启炫酷灯光效果教程
- 01-12Flipagram-Flipagram应用软件功能介绍
- 01-10SuperSlide插件介绍以及使用教程
- 02-23DedeCMS网站更换域名后文章图片路径批
- 07-18小米手环7pro息屏时长更改方法
- 07-14帝国CMS7.5版各系统模型新增记录信息
- 01-12微探-微探应用软件功能介绍
- 12-26圣诞画板-圣诞画板应用软件功能介绍
- 07-08html中iframe如何设置高度和宽度
- 10-12真我gtneo2怎么屏蔽广告
- 04-05详谈php ip2long 出现负数的原因及解
最近更新
阅读排行
猜你喜欢
- 02-19PhpWind教程:修改帖子标题颜色
- 10-15小米智能猫眼联网教程介绍
- 11-21MySql删除匿名用户的方法
- 04-01jQuery validate验证隐藏表单(hidden)
- 01-12应急119-应急119应用软件功能介绍
- 01-13秋葵编辑-秋葵编辑应用软件功能介绍
- 10-14支付宝怎么领养宠物
- 02-25小米音箱连接wifi网络操作方法
- 09-28抖音不可思议的照片3D运镜同款视频制
- 01-18ecolor涂色-ecolor涂色应用软件功能介