TensorFLow用Saver保存和恢复变量


Posted in Python onMarch 10, 2018

本文为大家分享了TensorFLow用Saver保存和恢复变量的具体代码,供大家参考,具体内容如下

建立文件tensor_save.py, 保存变量v1,v2的tensor到checkpoint files中,名称分别设置为v3,v4。

import tensorflow as tf

# Create some variables.
v1 = tf.Variable(3, name="v1")
v2 = tf.Variable(4, name="v2")

# Create model
y=tf.add(v1,v2)

# Add an op to initialize the variables.
init_op = tf.initialize_all_variables()

# Add ops to save and restore all the variables.
saver = tf.train.Saver({'v3':v1,'v4':v2})

# Later, launch the model, initialize the variables, do some work, save the
# variables to disk.
with tf.Session() as sess:
 sess.run(init_op)
 print("v1 = ", v1.eval())
 print("v2 = ", v2.eval())
 # Save the variables to disk.
 save_path = saver.save(sess, "f:/tmp/model.ckpt")
 print ("Model saved in file: ", save_path)

建立文件tensor_restror.py, 将checkpoint files中名称分别为v3,v4的tensor分别恢复到变量v3,v4中。

import tensorflow as tf

# Create some variables.
v3 = tf.Variable(0, name="v3")
v4 = tf.Variable(0, name="v4")

# Create model
y=tf.mul(v3,v4)

# Add ops to save and restore all the variables.
saver = tf.train.Saver()

# Later, launch the model, use the saver to restore variables from disk, and
# do some work with the model.
with tf.Session() as sess:
 # Restore variables from disk.
 saver.restore(sess, "f:/tmp/model.ckpt")
 print ("Model restored.")
 print ("v3 = ", v3.eval())
 print ("v4 = ", v4.eval())
 print ("y = ",sess.run(y))

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python计算圆周长、面积、球体体积并画出圆
Apr 08 Python
python批量提交沙箱问题实例
Oct 08 Python
Python内置函数—vars的具体使用方法
Dec 04 Python
python爬虫使用cookie登录详解
Dec 27 Python
解决seaborn在pycharm中绘图不出图的问题
May 24 Python
python: 自动安装缺失库文件的方法
Oct 22 Python
在matplotlib的图中设置中文标签的方法
Dec 13 Python
python Kmeans算法原理深入解析
Aug 23 Python
python多进程并行代码实例
Sep 30 Python
Python如何使用vars返回对象的属性列表
Oct 17 Python
django项目中使用云片网发送短信验证码的实现
Jan 19 Python
Python中全局变量和局部变量的理解与区别
Feb 07 Python
tensorflow创建变量以及根据名称查找变量
Mar 10 #Python
Python2中文处理纪要的实现方法
Mar 10 #Python
python实现冒泡排序算法的两种方法
Mar 10 #Python
Python使用pyh生成HTML文档的方法示例
Mar 10 #Python
tensorflow获取变量维度信息
Mar 10 #Python
TensorFlow变量管理详解
Mar 10 #Python
TensorFlow神经网络优化策略学习
Mar 09 #Python
You might like
PHP JSON 数据解析代码
2010/05/26 PHP
ThinkPHP中redirect用法分析
2014/12/05 PHP
php+ajax实时刷新简单实例
2015/02/25 PHP
PHP中使用socket方式GET、POST数据实例
2015/04/02 PHP
php使用post数组的键值创建同名变量并赋值的方法
2015/04/03 PHP
极酷的javascirpt,让你随意编辑任何网页
2007/02/25 Javascript
jquery连缀语法如何实现
2012/11/29 Javascript
利用JQuery和Servlet实现跨域提交请求示例分享
2014/02/12 Javascript
Node.js中的事件驱动编程详解
2014/08/16 Javascript
JavaScript Sort 的一个错误用法示例
2015/03/20 Javascript
js确认框confirm()用法实例详解
2016/01/07 Javascript
浏览器复制插件zeroclipboard使用指南
2016/03/26 Javascript
Javascript中的迭代、归并方法详解
2016/06/14 Javascript
AngularJS的依赖注入实例分析(使用module和injector)
2017/01/19 Javascript
jQuery DateTimePicker 日期和时间插件示例
2017/01/22 Javascript
vue项目中使用axios上传图片等文件操作
2017/11/02 Javascript
bootstrap-Treeview实现级联勾选
2017/11/23 Javascript
vue中eventbus被多次触发以及踩过的坑
2017/12/02 Javascript
解决linux下node.js全局模块找不到的问题
2018/05/15 Javascript
vue2.0 + ele的循环表单及验证字段方法
2018/09/18 Javascript
JS 正则表达式验证密码、邮箱格式的实例代码
2018/10/28 Javascript
浅谈在Vue.js中如何实现时间转换指令
2019/01/06 Javascript
JS原生瀑布流效果实现
2019/04/26 Javascript
使用vue打包进行云服务器上传的问题
2020/03/02 Javascript
Python实现的端口扫描功能示例
2018/04/08 Python
Python实现绘制双柱状图并显示数值功能示例
2018/06/23 Python
wxPython实现带颜色的进度条
2019/11/19 Python
英国标志性奢侈品牌:Burberry
2016/07/28 全球购物
娇韵诗Clarins意大利官方网站:法国天然护肤品牌
2020/03/11 全球购物
班组长安全工作职责
2014/07/15 职场文书
交通事故协议书范本
2014/11/18 职场文书
刘公岛导游词
2015/02/05 职场文书
体育部部长竞选稿
2015/11/21 职场文书
使用css样式设计一个简单的html登陆界面的实现
2021/03/30 HTML / CSS
sql通过日期判断年龄函数的示例代码
2021/07/16 SQL Server
MySQL添加索引特点及优化问题
2022/07/23 MySQL