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 Mysql自动备份脚本
Jul 14 Python
Python with用法实例
Apr 14 Python
Python中tell()方法的使用详解
May 24 Python
python中defaultdict的用法详解
Jun 07 Python
Python反射的用法实例分析
Feb 11 Python
Python利用正则表达式实现计算器算法思路解析
Apr 25 Python
python检测空间储存剩余大小和指定文件夹内存占用的实例
Jun 11 Python
使用numba对Python运算加速的方法
Oct 15 Python
python求最大值,不使用内置函数的实现方法
Jul 09 Python
django的ORM操作 删除和编辑实现详解
Jul 24 Python
Python使用lambda表达式对字典排序操作示例
Jul 25 Python
django中ImageField的使用详解
Dec 21 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
微信支付开发动态链接Native支付
2016/07/12 PHP
php取出数组单个值的方法
2018/03/12 PHP
PHP执行系统命令函数实例讲解
2021/03/03 PHP
Jquery 组合form元素为json格式,asp.net反序列化
2009/07/09 Javascript
jQuery Trim去除字符串首尾空字符的实现方法说明
2014/02/11 Javascript
node.js中的fs.renameSync方法使用说明
2014/12/16 Javascript
javascript比较两个日期的先后示例代码
2014/12/31 Javascript
jquery validate表单验证的基本用法入门
2016/01/18 Javascript
bootstrap布局中input输入框右侧图标点击功能
2016/05/16 Javascript
用JS动态改变表单form里的action值属性的两种方法
2016/05/25 Javascript
基于WebUploader的文件上传js插件
2016/08/19 Javascript
用jQuery.ajaxSetup实现对请求和响应数据的过滤
2016/12/20 Javascript
vue.js学习之UI组件开发教程
2017/07/03 Javascript
通俗解释JavaScript正则表达式快速记忆
2017/08/23 Javascript
快速处理vue渲染前的显示问题
2018/03/05 Javascript
使用vue中的v-for遍历二维数组的方法
2018/03/07 Javascript
vue-content-loader内容加载器的使用方法
2018/08/05 Javascript
Vue infinite update loop的问题解决
2019/04/23 Javascript
js验证身份证号码记录的方法
2019/04/26 Javascript
JS中数组实现代码(倒序遍历数组,数组连接字符串)
2019/12/29 Javascript
vue实现户籍管理系统
2020/05/29 Javascript
Python删除指定目录下过期文件的2个脚本分享
2014/04/10 Python
python批量从es取数据的方法(文档数超过10000)
2018/12/27 Python
css3实现二维码扫描特效的示例
2020/10/29 HTML / CSS
HTML5之SVG 2D入门9—蒙板及mask元素介绍与应用
2013/01/30 HTML / CSS
意大利时尚奢侈品店:D’Aniello Boutique
2021/01/19 全球购物
高级销售求职信
2014/02/21 职场文书
乡镇网格化管理实施方案
2014/03/23 职场文书
酒店仓管员岗位职责
2014/04/28 职场文书
标准版离职证明书
2014/09/12 职场文书
乡村教师党员四风问题对照检查材料思想汇报
2014/10/08 职场文书
2014年宣传思想工作总结
2014/12/10 职场文书
陈斌强事迹观后感
2015/06/17 职场文书
导游词之南京栖霞山
2019/10/18 职场文书
mysql全面解析json/数组
2022/07/07 MySQL
python如何将mat文件转为png
2022/07/15 Python