TensorFlow利用saver保存和提取参数的实例


Posted in Python onJuly 26, 2018

在训练循环中,定期调用 saver.save() 方法,向文件夹中写入包含了当前模型中所有可训练变量的 checkpoint 文件。

saver.save(sess, FLAGS.train_dir, global_step=step)

global_step是训练的第几步

保存参数:

import tensorflow as tf
 
W = tf.Variable([[1, 2, 3]], dtype=tf.float32)
b = tf.Variable([[1]], dtype=tf.float32)
 
saver = tf.train.Saver()
 
sess = tf.InteractiveSession()
tf.global_variables_initializer().run()
# 必须要指定文件夹,保存到ckpt文件
save_path = saver.save(sess, "winycg/1.ckpt")
print(save_path)

一次 saver.save() 后可以在文件夹中看到新增的四个文件,实际上每调用一次保存操作会创建后3个数据文件并创建一个检查点(checkpoint)文件,简单理解就是权重等参数被保存到 .chkp.data 文件中,以字典的形式;图和元数据被保存到 .chkp.meta 文件中,可以被 tf.train.import_meta_graph 加载到当前默认的图。

TensorFlow利用saver保存和提取参数的实例

读取参数:

import tensorflow as tf
import numpy as np
 
W = tf.Variable(np.arange(3).reshape(1, 3), dtype=tf.float32)
b = tf.Variable(np.arange(1).reshape(1, 1), dtype=tf.float32)
 
saver = tf.train.Saver()
 
sess = tf.InteractiveSession()
# 读取参数时不需要global_variables_initializer()
save_path = saver.restore(sess, "parameter/1.ckpt")
print("weights:", sess.run(W))
print("bias:", sess.run(b))

weights: [[ 1. 2. 3.]]

bias: [[ 1.]]

以上这篇TensorFlow利用saver保存和提取参数的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python获取指定目录下所有文件名列表的方法
May 20 Python
Python面向对象之接口、抽象类与多态详解
Aug 27 Python
Python使用post及get方式提交数据的实例
Jan 24 Python
详解django实现自定义manage命令的扩展
Aug 13 Python
在Python中使用MongoEngine操作数据库教程实例
Dec 03 Python
django商品分类及商品数据建模实例详解
Jan 03 Python
关于python pycharm中输出的内容不全的解决办法
Jan 10 Python
python软件都是免费的吗
Jun 18 Python
python实现图片转字符画的完整代码
Feb 21 Python
Django给表单添加honeypot验证增加安全性
May 06 Python
python实现简易自习室座位预约系统
Jun 30 Python
python中Pyqt5使用Qlabel标签播放视频
Apr 22 Python
78行Python代码实现现微信撤回消息功能
Jul 26 #Python
python opencv实现图片旋转矩形分割
Jul 26 #Python
Flask之flask-session的具体使用
Jul 26 #Python
tensorflow saver 保存和恢复指定 tensor的实例讲解
Jul 26 #Python
python opencv旋转图像(保持图像不被裁减)
Jul 26 #Python
详解Django中间件的5种自定义方法
Jul 26 #Python
python opencv实现切变换 不裁减图片
Jul 26 #Python
You might like
thinkphp分页实现效果
2016/10/13 PHP
JavaScript与Image加载事件(onload)、加载状态(complete)
2011/02/14 Javascript
如何将一个String和多个String值进行比较思路分析
2013/04/22 Javascript
JavaScript使用HTML5的window.postMessage实现跨域通信例子
2014/04/11 Javascript
让checkbox不选中即将选中的checkbox不选中
2014/07/11 Javascript
Javascript 拖拽的一些简单的应用(逐行分析代码,让你轻松了拖拽的原理)
2015/01/23 Javascript
innerHTML属性,outerHTML属性,textContent属性,innerText属性区别详解
2015/03/13 Javascript
js鼠标点击按钮切换图片-图片自动切换-点击左右按钮切换特效代码
2015/09/02 Javascript
bootstrap-treeview自定义双击事件实现方法
2016/01/09 Javascript
Angular 4.x中表单Reactive Forms详解
2017/04/25 Javascript
JavaScript实现的原生态Tab标签页功能【兼容IE6】
2017/09/18 Javascript
Angular自定义组件实现数据双向数据绑定的实例
2017/12/11 Javascript
详解nodejs解压版安装和配置(带有搭建前端项目脚手架)
2018/12/06 NodeJs
深入浅析vue-cli@3.0 使用及配置说明
2019/05/08 Javascript
解决vue单页面应用中动态修改title问题
2019/06/09 Javascript
Angular8路由守卫原理和使用方法
2019/08/29 Javascript
vue.js 实现a标签href里添加参数
2019/11/12 Javascript
[06:21]完美世界亚洲区首席发行官竺琦TI3采访
2013/08/26 DOTA
Python里隐藏的“禅”
2014/06/16 Python
python中requests模块的使用方法
2015/04/08 Python
使用python检测主机存活端口及检查存活主机
2015/10/12 Python
python 使用事件对象asyncio.Event来同步协程的操作
2020/05/04 Python
饿了么订餐官网:外卖、网上订餐
2019/06/28 全球购物
小学教师的自我评价范例
2013/10/31 职场文书
心理学专业毕业生推荐信范文
2013/11/21 职场文书
大学本科毕业生的自我鉴定
2013/11/26 职场文书
财务会计实习报告体会
2013/12/20 职场文书
写给保洁员表扬信
2014/01/08 职场文书
社区工作者先进事迹
2014/01/18 职场文书
微观物理专业自荐信
2014/01/26 职场文书
护士自我评价
2014/02/01 职场文书
中专自我鉴定
2014/02/05 职场文书
幼儿园母亲节活动方案
2014/03/10 职场文书
公证委托书格式
2014/09/13 职场文书
导游词之丹东鸭绿江
2019/10/24 职场文书
解决Goland 同一个package中函数互相调用的问题
2021/05/06 Golang