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 相关文章推荐
10个易被忽视但应掌握的Python基本用法
Apr 01 Python
分享Python开发中要注意的十个小贴士
Aug 30 Python
Python模拟登录的多种方法(四种)
Jun 01 Python
Pandas读取MySQL数据到DataFrame的方法
Jul 25 Python
python安装pywin32clipboard的操作方法
Jan 24 Python
Django页面数据的缓存与使用的具体方法
Apr 23 Python
python3 动态模块导入与全局变量使用实例
Dec 22 Python
python 实现屏幕录制示例
Dec 23 Python
QML实现钟表效果
Jun 02 Python
sklearn的predict_proba使用说明
Jun 28 Python
Django使用echarts进行可视化展示的实践
Jun 10 Python
Python中rapidjson参数校验实现
Jul 25 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
杏林同学录(五)
2006/10/09 PHP
一个漂亮的php验证码类(分享)
2013/08/06 PHP
php获取网站百度快照日期的方法
2015/07/29 PHP
php对文件夹进行相关操作(遍历、计算大小)
2015/11/04 PHP
抛弃 PHP 代价太高
2016/04/26 PHP
PHP实现的权重算法示例【可用于游戏根据权限来随机物品】
2019/02/15 PHP
如何实现JS函数的重载
2006/09/22 Javascript
javascript OFFICE控件测试代码
2009/12/08 Javascript
extjs表格文本启用选择复制功能具体实现
2013/10/11 Javascript
动态添加删除表格行的js实现代码
2014/02/28 Javascript
js实现的标题栏新消息闪烁提示效果
2014/06/06 Javascript
JavaScript设计模式之建造者模式介绍
2014/12/28 Javascript
分享一个常用的javascript静态类
2014/12/31 Javascript
JavaScript针对网页节点的增删改查用法实例
2015/02/02 Javascript
javascript+ajax实现产品页面加载信息
2015/07/09 Javascript
javascript工厂模式和构造函数模式创建对象方法解析
2016/12/30 Javascript
原生JS实现垂直手风琴效果
2017/02/19 Javascript
Bootstrap响应式表格详解
2017/05/23 Javascript
Express的HTTP重定向到HTTPS的方法
2018/06/06 Javascript
微信小程序项目实践之九宫格实现及item跳转功能
2018/07/19 Javascript
node上的redis调用优化示例详解
2018/10/30 Javascript
详解Vue中使用插槽(slot)、聚类插槽
2019/04/12 Javascript
jQuery与原生JavaScript选择HTML元素集合用法对比分析
2019/11/26 jQuery
python实现的登陆Discuz!论坛通用代码分享
2014/07/11 Python
Python中的作用域规则详解
2015/01/30 Python
Python进程间通信 multiProcessing Queue队列实现详解
2019/09/23 Python
Django Auth用户认证组件实现代码
2020/10/13 Python
阿联酋优惠券服务:Living Kool
2019/12/12 全球购物
创业资金计划书
2014/02/06 职场文书
国培教师自我鉴定
2014/02/12 职场文书
《陈涉世家》教学反思
2014/04/12 职场文书
2015庆祝七一建党节94周年活动总结
2015/03/20 职场文书
学校元旦晚会开场白
2015/05/29 职场文书
企业版Windows 11有哪些新功能? Win11适用于企业的功能介绍
2021/11/21 数码科技
java开发双人五子棋游戏
2022/05/06 Java/Android
正则表达式基础与常用验证表达式
2022/06/16 Javascript