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之编写类之四再论继承
Oct 11 Python
Python实现求笛卡尔乘积的方法
Sep 16 Python
简单了解OpenCV是个什么东西
Nov 10 Python
Python实现PS滤镜碎片特效功能示例
Jan 24 Python
python如何读写json数据
Mar 21 Python
Python爬取成语接龙类网站
Oct 19 Python
Python将8位的图片转为24位的图片实现方法
Oct 24 Python
Python 类方法和实例方法(@classmethod),静态方法(@staticmethod)原理与用法分析
Sep 20 Python
pytorch中的自定义反向传播,求导实例
Jan 06 Python
opencv 形态学变换(开运算,闭运算,梯度运算)
Jul 07 Python
python实现sm2和sm4国密(国家商用密码)算法的示例
Sep 26 Python
Python基于Serializer实现字段验证及序列化
Nov 04 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
一篇入门的php Class 文章
2007/04/04 PHP
php数据库密码的找回的步骤
2011/01/12 PHP
thinkPHP中分页用法实例分析
2015/12/26 PHP
无需数据库在线投票调查php代码
2016/07/20 PHP
PHP7引入的"??"和"?:"的区别讲解
2019/04/08 PHP
javascript 对象的定义方法
2007/01/10 Javascript
JavaScript.The.Good.Parts阅读笔记(一)假值与===运算符
2010/11/16 Javascript
javascript中日期转换成时间戳的小例子
2013/03/21 Javascript
JavaScript中DOM详解
2015/04/13 Javascript
JavaScript中几种排序算法的简单实现
2015/07/29 Javascript
JavaScript 保护变量不被随意修改的实现代码
2017/09/27 Javascript
js实现数组内数据的上移和下移的实例
2017/11/14 Javascript
laravel5.3 vue 实现收藏夹功能实例详解
2018/01/21 Javascript
Javascript数组方法reduce的妙用之处分享
2019/06/10 Javascript
150行Node.js实现的dns代理工具
2019/08/02 Javascript
Element MessageBox弹框的具体使用
2020/07/27 Javascript
[02:12]打造更好的电竞完美世界:完美盛典回顾篇
2018/12/19 DOTA
在Python的Flask中使用WTForms表单框架的基础教程
2016/06/07 Python
python实现决策树C4.5算法详解(在ID3基础上改进)
2017/05/31 Python
Python编程实现及时获取新邮件的方法示例
2017/08/10 Python
python爱心表白 每天都是浪漫七夕!
2018/08/18 Python
python try 异常处理(史上最全)
2019/03/07 Python
python自制包并用pip免提交到pypi仅安装到本机【推荐】
2019/06/03 Python
python pandas时序处理相关功能详解
2019/07/03 Python
Pandas时间序列:重采样及频率转换方式
2019/12/26 Python
Python图像处理库PIL的ImageDraw模块介绍详解
2020/02/26 Python
Python 随机生成测试数据的模块:faker基本使用方法详解
2020/04/09 Python
python 实现rolling和apply函数的向下取值操作
2020/06/08 Python
Python基于pillow库实现生成图片水印
2020/09/14 Python
CSS3 实现的火焰动画
2020/12/07 HTML / CSS
通过HTML5规范搞定i、em、b、strong元素的区别
2017/03/04 HTML / CSS
伯克斯奥特莱斯:Burkes Outlet
2019/03/30 全球购物
杭州时比特电子有限公司SQL
2013/08/22 面试题
英文自荐信
2013/12/15 职场文书
考教师资格证不要错过的4个最佳时机
2019/07/17 职场文书
Python通用验证码识别OCR库ddddocr的安装使用教程
2022/07/07 Python