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使用函数默认值实现函数静态变量的方法
Aug 18 Python
Python获取网页上图片下载地址的方法
Mar 11 Python
Python+Pika+RabbitMQ环境部署及实现工作队列的实例教程
Jun 29 Python
pandas 使用apply同时处理两列数据的方法
Apr 20 Python
Python中实现变量赋值传递时的引用和拷贝方法
Apr 29 Python
tensorflow更改变量的值实例
Jul 30 Python
python3 json数据格式的转换(dumps/loads的使用、dict to str/str to dict、json字符串/字典的相互转换)
Apr 01 Python
python实现屏保程序(适用于背单词)
Jul 30 Python
python实现门限回归方式
Feb 29 Python
Python统计学一数据的概括性度量详解
Mar 03 Python
python计算auc的方法
Sep 09 Python
Python类的继承super相关原理解析
Oct 22 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删除数组中特定元素的两种方法
2013/07/02 PHP
php将session放入memcached的设置方法
2014/02/14 PHP
PHP实现八皇后算法
2019/05/06 PHP
学习ExtJS(二) Button常用方法
2009/10/07 Javascript
jQuery浏览器CSS3特写兼容实例
2015/01/19 Javascript
第八篇Bootstrap下拉菜单实例代码
2016/06/21 Javascript
数据结构中的各种排序方法小结(JS实现)
2016/07/23 Javascript
AngularJS入门教程之过滤器用法示例
2016/11/02 Javascript
vuejs 单文件组件.vue 文件的使用
2017/07/28 Javascript
Es6 Generator函数详细解析
2018/02/24 Javascript
Vue数据监听方法watch的使用
2018/03/28 Javascript
JQuery中queue方法用法示例
2019/01/31 jQuery
使用ESLint禁止项目导入特定模块的方法步骤
2019/03/04 Javascript
浅析vue-cli3配置webpack-bundle-analyzer插件【推荐】
2019/10/23 Javascript
基于better-scroll 实现歌词联动功能的代码
2020/05/07 Javascript
详解Python判定IP地址合法性的三种方法
2018/03/06 Python
python读取TXT每行,并存到LIST中的方法
2018/10/26 Python
Python开启线程,在函数中开线程的实例
2019/02/22 Python
在OpenCV里使用Camshift算法的实现
2019/11/22 Python
Tensorflow轻松实现XOR运算的方式
2020/02/03 Python
通过python检测字符串的字母
2020/02/18 Python
探秘TensorFlow 和 NumPy 的 Broadcasting 机制
2020/03/13 Python
英国领先的游戏零售商:GAME
2019/09/24 全球购物
怎样让char类型的东西转换成int类型
2013/12/09 面试题
校园文明倡议书
2014/05/16 职场文书
幼儿园保育员责任书
2014/07/22 职场文书
销售人员求职信
2014/07/22 职场文书
四风问题对照检查整改措施思想报告
2014/10/05 职场文书
房地产财务经理岗位职责
2015/04/08 职场文书
专项资金申请报告
2015/05/15 职场文书
就业推荐表院系意见
2015/06/05 职场文书
团结友爱主题班会
2015/08/13 职场文书
2016年小学植树节活动总结
2016/03/16 职场文书
关于SpringBoot 使用 Redis 分布式锁解决并发问题
2021/11/17 Redis
Rust 连接 PostgreSQL 数据库的详细过程
2022/01/22 PostgreSQL
漫画「古见同学有交流障碍症」第25卷封面公开
2022/03/21 日漫