TensorFlow Saver:保存和读取模型参数.ckpt实例


Posted in Python onFebruary 10, 2020

在使用TensorFlow的过程中,保存模型参数变量是很重要的一个环节,既可以保证训练过程信息不丢失,也可以帮助我们在需要快速恢复或使用一个模型的时候,利用之前保存好的参数之间导入,可以节省大量的训练时间。本文通过最简单的例程教大家如何保存和读取.ckpt文件。

一、保存到文件

首先是导入必要的东西:

import tensorflow as tf
import numpy as np

随便写几个变量:

# Save to file
# remember to define the same dtype and shape when restore
W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weights')
b = tf.Variable([[1,2,3]], dtype=tf.float32, name='biases')
 
init= tf.initialize_all_variables()

定义一个saver,来存储我们的各种变量:

saver = tf.train.Saver()

保存的文件用.ckpt后缀:

with tf.Session() as sess:
  sess.run(init)
  save_path = saver.save(sess, "my_net/save_net.ckpt")
  print("Save to path: ", save_path)

上面我们就完成了保存操作。

接下来我们要把之前保存过的变量取出来。

二、取出之前保存的变量

这里要注意,取出时要先开辟一个容器来装,shape和type要和我们之前保存的.ckpt一样。

# restore variables
# redefine the same shape and same type for your variables
W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")
b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")

restore时,不需要进行init= tf.initialize_all_variables()操作。

利用saver提取文件:

saver = tf.train.Saver()
with tf.Session() as sess:
  saver.restore(sess, "my_net/save_net.ckpt")
  print("weights:", sess.run(W))
  print("biases:", sess.run(b))

结果:

TensorFlow Saver:保存和读取模型参数.ckpt实例

以上这篇TensorFlow Saver:保存和读取模型参数.ckpt实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python和shell变量互相传递的几种方法
Nov 20 Python
java直接调用python脚本的例子
Feb 16 Python
Python3实现从文件中读取指定行的方法
May 22 Python
Python 如何访问外围作用域中的变量
Sep 11 Python
Python实现图片转字符画的示例代码
Aug 21 Python
python中lambda()的用法
Nov 16 Python
Django实现组合搜索的方法示例
Jan 23 Python
Tensorflow实现卷积神经网络用于人脸关键点识别
Mar 05 Python
Django+zTree构建组织架构树的方法
Aug 21 Python
python批量将excel内容进行翻译写入功能
Oct 10 Python
手把手教你Python yLab的绘制折线图的画法
Oct 23 Python
Python numpy多维数组实现原理详解
Mar 10 Python
tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader
Feb 10 #Python
解决tensorflow添加ptb库的问题
Feb 10 #Python
使用TensorFlow直接获取处理MNIST数据方式
Feb 10 #Python
python如何实现单链表的反转
Feb 10 #Python
基于python求两个列表的并集.交集.差集
Feb 10 #Python
python实现串口通信的示例代码
Feb 10 #Python
基于Tensorflow高阶读写教程
Feb 10 #Python
You might like
PHP之浮点数计算比较以及取整数不准确的解决办法
2015/07/29 PHP
thinkphp多表查询两表有重复相同字段的完美解决方法
2016/09/22 PHP
Thinkphp3.2实用篇之计算型验证码示例
2017/02/09 PHP
PHP分页显示的方法分析【附PHP通用分页类】
2018/05/10 PHP
PHP For循环字母A-Z当超过26个字母时输出AA,AB,AC
2020/02/16 PHP
JavaScript 变量基础知识
2009/11/07 Javascript
BOM与DOM的区别分析
2010/10/26 Javascript
jQuery JSON的解析方式分享
2011/04/05 Javascript
用jquery实现动画跳到顶部和底部(这个比较简单)
2014/09/01 Javascript
详细解读AngularJS中的表单验证编程
2015/06/19 Javascript
微信小程序  action-sheet详解及实例代码
2016/11/09 Javascript
JavaScript动态绑定详解
2017/09/14 Javascript
jQuery实现仿京东防抖动菜单效果示例
2018/07/06 jQuery
浅谈nuxtjs校验登录中间件和混入(mixin)
2020/11/06 Javascript
Nuxt 项目性能优化调研分析
2020/11/07 Javascript
js前端对于大量数据的展示方式及处理方法
2020/12/02 Javascript
使用XML库的方式,实现RPC通信的方法(推荐)
2017/06/14 Python
浅析Git版本控制器使用
2017/12/10 Python
python模块smtplib实现纯文本邮件发送功能
2018/05/22 Python
python如何调用java类
2020/07/05 Python
Python3 用什么IDE开发工具比较好
2020/11/28 Python
详解通过focusout事件解决IOS键盘收起时界面不归位的问题
2019/07/18 HTML / CSS
公司JAVA开发面试题
2015/04/02 面试题
成人大专自我鉴定范文
2013/10/19 职场文书
中专生求职自荐信范文
2013/12/22 职场文书
施工资料员岗位职责
2014/01/06 职场文书
物流毕业生个人的自我评价
2014/02/13 职场文书
食品安全标语
2014/06/07 职场文书
服务理念口号
2014/06/11 职场文书
怎样写离婚协议书
2015/01/26 职场文书
委托书格式范文
2015/01/28 职场文书
幼儿园父亲节活动总结
2015/02/12 职场文书
2015年平安创建工作总结
2015/04/29 职场文书
学校勤俭节约倡议书
2015/04/29 职场文书
2015年度个人工作总结报告
2015/10/24 职场文书
Java中PriorityQueue实现最小堆和最大堆的用法
2021/06/27 Java/Android