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 相关文章推荐
Python2.x中文乱码问题解决方法
Jun 02 Python
django 发送手机验证码的示例代码
Apr 25 Python
Django学习教程之静态文件的调用详解
May 08 Python
django从请求到响应的过程深入讲解
Aug 01 Python
用pycharm开发django项目示例代码
Oct 24 Python
python pyheatmap包绘制热力图
Nov 09 Python
通过python爬虫赚钱的方法
Jan 29 Python
Python使用Pandas库实现MySQL数据库的读写
Jul 06 Python
使用批处理脚本自动生成并上传NuGet包(操作方法)
Nov 19 Python
Python语法垃圾回收机制原理解析
Mar 25 Python
python 代码实现k-means聚类分析的思路(不使用现成聚类库)
Jun 01 Python
Python自定义sorted排序实现方法详解
Sep 18 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脚本加密专家php解密算法
2020/09/13 PHP
php中OR与|| AND与&&的区别总结
2013/10/26 PHP
Laravel执行migrate命令提示:No such file or directory的解决方法
2016/03/16 PHP
PHP的时间戳与具体时间转化的简单实现
2016/06/13 PHP
php读取和保存base64编码的图片内容
2017/04/22 PHP
使用PHP json_decode可能遇到的坑与解决方法
2017/08/03 PHP
TP5框架实现上传多张图片的方法分析
2020/03/29 PHP
TFDN图片播放器 不错自动播放
2006/10/03 Javascript
在js中单选框和复选框获取值的方式
2009/11/06 Javascript
jQuery 获取URL参数的插件
2010/03/04 Javascript
基于JQuery的模拟苹果桌面Dock效果(稳定版)
2012/10/15 Javascript
js判断两个日期是否相等的方法
2013/09/10 Javascript
JS转换HTML转义符的方法
2016/08/24 Javascript
JavaScript基于自定义函数判断变量类型的实现方法
2016/11/23 Javascript
Bootstrap复选框和单选按钮美化插件(推荐)
2016/11/23 Javascript
基于vue.js实现图片轮播效果
2016/12/01 Javascript
简单实现Bootstrap标签页
2020/08/09 Javascript
详解webpack 多入口配置
2017/06/16 Javascript
微信小程序实现搜索功能并跳转搜索结果页面
2019/05/18 Javascript
vue keep-alive列表页缓存 详情页返回上一页不刷新,定位到之前位置
2019/11/26 Javascript
[01:23]2014DOTA2国际邀请赛 球迷无处不在Ti现场世界杯受关注
2014/07/10 DOTA
Python 初始化多维数组代码
2008/09/06 Python
Pandas实现数据类型转换的一些小技巧汇总
2018/05/07 Python
对numpy中数组转置的求解以及向量内积计算方法
2018/10/31 Python
python 类之间的参数传递方式
2019/12/20 Python
Python GUI库PyQt5样式QSS子控件介绍
2020/02/25 Python
HTML5验证以及日期显示的实现详解
2013/07/05 HTML / CSS
越南综合购物网站:Lazada越南
2019/06/10 全球购物
德国购买踏板车网站:Microscooter
2019/10/14 全球购物
石油工程专业毕业生求职信
2014/04/13 职场文书
2014年教师党员公开承诺书
2014/05/28 职场文书
2014法制宣传日活动总结
2014/07/09 职场文书
2014年个人思想工作总结
2014/11/27 职场文书
典型事迹材料范文
2014/12/29 职场文书
社区服务理念口号
2015/12/25 职场文书
使用CSS3实现按钮悬停闪烁动态特效代码
2021/08/30 HTML / CSS