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合并文本文件示例
Feb 07 Python
python 远程统计文件代码分享
May 14 Python
Python中使用装饰器来优化尾递归的示例
Jun 18 Python
详解使用python crontab设置linux定时任务
Dec 08 Python
使用PyV8在Python爬虫中执行js代码
Feb 16 Python
浅析Python中的赋值和深浅拷贝
Aug 15 Python
利用Django-environ如何区分不同环境
Aug 26 Python
自学python的建议和周期预算
Jan 30 Python
python输出带颜色字体实例方法
Sep 01 Python
TensorFlow的reshape操作 tf.reshape的实现
Apr 19 Python
Linux安装Python3如何和系统自带的Python2并存
Jul 23 Python
在Ubuntu中安装并配置Pycharm教程的实现方法
Jan 06 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
xml在joomla表单中的应用详解分享
2012/07/19 PHP
PHP系列学习之日期函数使用介绍
2012/08/18 PHP
laravel 之 Eloquent 模型修改器和序列化示例
2019/10/17 PHP
PHP设计模式(一)工厂模式Factory实例详解【创建型】
2020/05/02 PHP
解决FireFox下[使用event很麻烦]的问题
2006/11/26 Javascript
Javascript string 扩展库代码
2010/04/09 Javascript
jQuery:delegate中select()不起作用的解决方法(实例讲解)
2014/01/26 Javascript
js中substring和substr的定义和用法
2014/05/05 Javascript
jQuery实现本地预览上传图片功能
2016/01/08 Javascript
Bootstrap 轮播(Carousel)插件
2016/12/26 Javascript
微信小程序 数据遍历的实现
2017/04/05 Javascript
Node使用Sequlize连接Mysql报错:Access denied for user ‘xxx’@‘localhost’
2018/01/03 Javascript
vue-infinite-loading2.0 中文文档详解
2018/04/08 Javascript
详解关于html,css,js三者的加载顺序问题
2019/04/10 Javascript
js数组相减简单示例【删除a数组所有与b数组相同元素】
2020/03/04 Javascript
在Echarts图中给坐标轴加一个标识线markLine
2020/07/20 Javascript
vue深度监听(监听对象和数组的改变)与立即执行监听实例
2020/09/04 Javascript
[51:53]完美世界DOTA2联赛决赛日 Inki vs LBZS 第二场 11.08
2020/11/10 DOTA
Python使用multiprocessing创建进程的方法
2015/06/04 Python
Python3编码问题 Unicode utf-8 bytes互转方法
2018/10/26 Python
python不使用for计算两组、多个矩形两两间的iou方式
2020/01/18 Python
django 将自带的数据库sqlite3改成mysql实例
2020/07/09 Python
基于CSS3的animation属性实现微信拍一拍动画效果
2020/06/22 HTML / CSS
英国标志性奢侈品牌:Burberry
2016/07/28 全球购物
妈妈的账单教学反思
2014/02/06 职场文书
公益广告宣传方案
2014/02/28 职场文书
商务日语专业毕业生自荐信
2014/03/27 职场文书
学生干部培训方案
2014/06/12 职场文书
爱护草坪标语
2014/06/24 职场文书
最美家庭活动方案
2014/08/31 职场文书
教师批评与自我批评发言稿
2014/10/15 职场文书
开展党的群众路线教育实践活动总结报告
2014/10/31 职场文书
2015国际残疾人日活动总结
2015/03/24 职场文书
2015年幼儿园学期工作总结
2015/05/22 职场文书
大学生暑假实习总结
2015/07/13 职场文书
把77A收信机改造成收音机
2022/04/05 无线电