Tensorflow之Saver的用法详解


Posted in Python onApril 23, 2018

Saver的用法

1. Saver的背景介绍

我们经常在训练完一个模型之后希望保存训练的结果,这些结果指的是模型的参数,以便下次迭代的训练或者用作测试。Tensorflow针对这一需求提供了Saver类。

Saver类提供了向checkpoints文件保存和从checkpoints文件中恢复变量的相关方法。Checkpoints文件是一个二进制文件,它把变量名映射到对应的tensor值 。

只要提供一个计数器,当计数器触发时,Saver类可以自动的生成checkpoint文件。这让我们可以在训练过程中保存多个中间结果。例如,我们可以保存每一步训练的结果。

为了避免填满整个磁盘,Saver可以自动的管理Checkpoints文件。例如,我们可以指定保存最近的N个Checkpoints文件。

2. Saver的实例

下面以一个例子来讲述如何使用Saver类 

import tensorflow as tf 
import numpy as np  
x = tf.placeholder(tf.float32, shape=[None, 1]) 
y = 4 * x + 4  
w = tf.Variable(tf.random_normal([1], -1, 1)) 
b = tf.Variable(tf.zeros([1])) 
y_predict = w * x + b 
loss = tf.reduce_mean(tf.square(y - y_predict)) 
optimizer = tf.train.GradientDescentOptimizer(0.5) 
train = optimizer.minimize(loss)  
isTrain = False 
train_steps = 100 
checkpoint_steps = 50 
checkpoint_dir = ''  
saver = tf.train.Saver() # defaults to saving all variables - in this case w and b 
x_data = np.reshape(np.random.rand(10).astype(np.float32), (10, 1))  
with tf.Session() as sess: 
  sess.run(tf.initialize_all_variables()) 
  if isTrain: 
    for i in xrange(train_steps): 
      sess.run(train, feed_dict={x: x_data}) 
      if (i + 1) % checkpoint_steps == 0: 
        saver.save(sess, checkpoint_dir + 'model.ckpt', global_step=i+1) 
  else: 
    ckpt = tf.train.get_checkpoint_state(checkpoint_dir) 
    if ckpt and ckpt.model_checkpoint_path: 
      saver.restore(sess, ckpt.model_checkpoint_path) 
    else: 
      pass 
    print(sess.run(w)) 
    print(sess.run(b))
  1. isTrain:用来区分训练阶段和测试阶段,True表示训练,False表示测试
  2. train_steps:表示训练的次数,例子中使用100
  3. checkpoint_steps:表示训练多少次保存一下checkpoints,例子中使用50
  4. checkpoint_dir:表示checkpoints文件的保存路径,例子中使用当前路径

2.1 训练阶段

使用Saver.save()方法保存模型:

  1. sess:表示当前会话,当前会话记录了当前的变量值
  2. checkpoint_dir + 'model.ckpt':表示存储的文件名
  3. global_step:表示当前是第几步

训练完成后,当前目录底下会多出5个文件。

Tensorflow之Saver的用法详解

打开名为“checkpoint”的文件,可以看到保存记录,和最新的模型存储位置。

Tensorflow之Saver的用法详解

2.1测试阶段

测试阶段使用saver.restore()方法恢复变量:

sess:表示当前会话,之前保存的结果将被加载入这个会话

ckpt.model_checkpoint_path:表示模型存储的位置,不需要提供模型的名字,它会去查看checkpoint文件,看看最新的是谁,叫做什么。

运行结果如下图所示,加载了之前训练的参数w和b的结果

Tensorflow之Saver的用法详解

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

Python 相关文章推荐
linux系统使用python监测系统负载脚本分享
Jan 15 Python
Python的Django框架中从url中捕捉文本的方法
Jul 20 Python
Python cookbook(数据结构与算法)将名称映射到序列元素中的方法
Mar 22 Python
Python退火算法在高次方程的应用
Jul 26 Python
django小技巧之html模板中调用对象属性或对象的方法
Nov 30 Python
对python 命令的-u参数详解
Dec 03 Python
Python数据可视化教程之Matplotlib实现各种图表实例
Jan 13 Python
python中update的基本使用方法详解
Jul 17 Python
python mqtt 客户端的实现代码实例
Sep 25 Python
python 浅谈serial与stm32通信的编码问题
Dec 18 Python
python3中确保枚举值代码分析
Dec 02 Python
python基于pexpect库自动获取日志信息
Feb 01 Python
python获取文件路径、文件名、后缀名的实例
Apr 23 #Python
Python基于FTP模块实现ftp文件上传操作示例
Apr 23 #Python
Python基于whois模块简单识别网站域名及所有者的方法
Apr 23 #Python
Python实现自定义顺序、排列写入数据到Excel的方法
Apr 23 #Python
python 获取指定文件夹下所有文件名称并写入列表的实例
Apr 23 #Python
python遍历一个目录,输出所有的文件名的实例
Apr 23 #Python
python 获取文件下所有文件或目录os.walk()的实例
Apr 23 #Python
You might like
细谈php中SQL注入攻击与XSS攻击
2012/06/10 PHP
PHP生成随机用户名和密码的实现代码
2013/02/27 PHP
ThinkPHP3.2框架使用addAll()批量插入数据的方法
2017/03/16 PHP
PHP实现链式操作的三种方法详解
2017/11/16 PHP
javascript 关于# 和 void的区别分析
2009/10/26 Javascript
Javascript 面向对象 命名空间
2010/05/13 Javascript
jQuery中:text选择器用法实例
2015/01/03 Javascript
jQuery插件简单学习实例教程
2016/07/01 Javascript
深入剖析Express cookie-parser中间件实现示例
2018/02/01 Javascript
把vue-router和express项目部署到服务器的方法
2018/02/21 Javascript
微信小程序 JS动态修改样式的实现方法
2018/12/16 Javascript
浅谈redux, koa, express 中间件实现对比解析
2019/05/23 Javascript
微信小程序Echarts覆盖正常组件问题解决
2019/07/13 Javascript
Vuex实现数据增加和删除功能
2019/11/11 Javascript
Vue中的this.$options.data()和this.$data用法说明
2020/07/26 Javascript
详解JavaScript中new操作符的解析和实现
2020/09/04 Javascript
解决vue 使用axios.all()方法发起多个请求控制台报错的问题
2020/11/09 Javascript
浅谈vue在html中出现{{}}的原因及解决方式
2020/11/16 Javascript
[01:45]DOTA2众星出演!DSPL刀塔次级职业联赛宣传片
2014/11/21 DOTA
Django实现登录随机验证码的示例代码
2018/06/20 Python
linux环境中没有网络怎么下载python
2019/07/07 Python
python爬虫 线程池创建并获取文件代码实例
2019/09/28 Python
解决pycharm启动后总是不停的updating indices...indexing的问题
2019/11/27 Python
Html5 web本地存储实例详解
2016/07/28 HTML / CSS
销售高级职员求职信
2013/10/29 职场文书
办公室驾驶员岗位职责
2013/11/15 职场文书
自我鉴定怎么写
2013/12/05 职场文书
拾金不昧表扬信范文
2014/01/11 职场文书
建议书怎么写
2014/03/12 职场文书
安全生产月宣传标语
2014/10/06 职场文书
2014年加油站工作总结
2014/12/04 职场文书
给老婆道歉的话
2015/01/20 职场文书
学校办公室主任岗位职责
2015/04/01 职场文书
三国演义读书笔记
2015/06/25 职场文书
2016年小学感恩节活动总结
2016/04/01 职场文书
Python编写nmap扫描工具
2021/07/21 Python