tensorflow实现训练变量checkpoint的保存与读取


Posted in Python onFebruary 10, 2020

1.保存变量

先创建(在tf.Session()之前)saver

saver = tf.train.Saver(tf.global_variables(),max_to_keep=1)  #max_to_keep这个保证只保存最后一次training的训练数据

然后在训练的循环里面

checkpoint_path = os.path.join(Path, 'model.ckpt') saver.save(session, checkpoint_path, global_step=step) #这里的step是循环训练的次数,也就是第几次迭代

以下保存的变量文件

tensorflow实现训练变量checkpoint的保存与读取

2.变量读取

1.若要直接恢复所有变量可以

saver = tf.train.Saver(tf.global_variables())
moudke_file=tf.train.latest_checkpoint('PATH')
saver.restore(sess,moudke_file)

PATH是存放保存变量的路径,会自动找到最近保存的变量文件

2 若想读取其中一部分变量值

def read_checkpoint():
  w = []
  checkpoint_path = '/home/ximao/models/resnet3/variable_logs/model.ckpt-17000'
  reader = tf.train.NewCheckpointReader(checkpoint_path)
  var = reader.get_variable_to_shape_map()
  for key in var:
    if 'weights' in key and 'conv' in key and 'Mo' not in key:
      print('tensorname:', key)
  #   # print(reader.get_tensor(key))

3. 若想恢复其中一部分变量值到新网络

(1)首先你要先获取你想要赋值新网络变量的变量名,这里变量名不是一个字符串,而是<name,shape,dtype>这样的一个结构,

然后把你要赋值的元素转为张量,最后把值赋给你得到变量名 如下:

var=[v for v in weight_pruned if v.op.name=='WRN/conv1/weights']
conv1_temp=tf.convert_to_tensor(conv1,dtype=tf.float32)
sess.run(tf.assign(var[0],conv1_temp))

weight_pruned 存放的是你新网络中所有的变量

以上这篇tensorflow实现训练变量checkpoint的保存与读取就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中关于使用模块的基础知识
May 24 Python
Python下实现的RSA加密/解密及签名/验证功能示例
Jul 17 Python
Python3实现抓取javascript动态生成的html网页功能示例
Aug 22 Python
Python数据结构与算法之图的最短路径(Dijkstra算法)完整实例
Dec 12 Python
python模块smtplib实现纯文本邮件发送功能
May 22 Python
Python实现模拟浏览器请求及会话保持操作示例
Jul 30 Python
influx+grafana自定义python采集数据和一些坑的总结
Sep 17 Python
Django+JS 实现点击头像即可更改头像的方法示例
Dec 26 Python
python队列Queue的详解
May 10 Python
详解Python 中sys.stdin.readline()的用法
Sep 12 Python
PyTorch 解决Dataset和Dataloader遇到的问题
Jan 08 Python
Django如何使用jwt获取用户信息
Apr 21 Python
Python单元测试模块doctest的具体使用
Feb 10 #Python
简单了解django文件下载方式
Feb 10 #Python
TensorFlow Saver:保存和读取模型参数.ckpt实例
Feb 10 #Python
tensorflow实现读取模型中保存的值 tf.train.NewCheckpointReader
Feb 10 #Python
解决tensorflow添加ptb库的问题
Feb 10 #Python
使用TensorFlow直接获取处理MNIST数据方式
Feb 10 #Python
python如何实现单链表的反转
Feb 10 #Python
You might like
phpBB BBcode处理的漏洞
2006/10/09 PHP
PHP+javascript液晶时钟
2006/10/09 PHP
jQuery 页面载入进度条实现代码
2009/02/08 Javascript
javascript学习笔记(七)利用javascript来创建和存储cookie
2011/04/08 Javascript
Javascript继承(上)——对象构建介绍
2012/11/08 Javascript
jquery复选框全选/取消示例
2013/12/30 Javascript
JS烟花背景效果实现方法
2015/03/03 Javascript
JS 作用域与作用域链详解
2015/04/07 Javascript
javascript中offset、client、scroll的属性总结
2015/08/13 Javascript
JavaScript中SetInterval与setTimeout的用法详解
2015/11/10 Javascript
深入理解JavaScript程序中内存泄漏
2016/03/17 Javascript
关于javascript原型的修改与重写(覆盖)差别详解
2016/08/31 Javascript
JS生成一维码(条形码)功能示例
2017/01/19 Javascript
解决jquery appaend元素中id绑定事件失效的问题
2017/09/12 jQuery
JavaScript 自定义事件之我见
2017/09/25 Javascript
JavaScript实现音乐自动切换和轮播
2017/11/05 Javascript
vue 的点击事件获取当前点击的元素方法
2018/09/15 Javascript
layui-table表复选框勾选的所有行数据获取的例子
2019/09/13 Javascript
Python警察与小偷的实现之一客户端与服务端通信实例
2014/10/09 Python
在Windows服务器下用Apache和mod_wsgi配置Python应用的教程
2015/05/06 Python
对python-3-print重定向输出的几种方法总结
2018/05/11 Python
pygame实现雷电游戏雏形开发
2018/11/20 Python
python wxpython 实现界面跳转功能
2019/12/17 Python
Python内置数据类型list各方法的性能测试过程解析
2020/01/07 Python
Python根据指定文件生成XML的方法
2020/06/29 Python
纯CSS3制作页面切换效果的实例代码
2019/05/30 HTML / CSS
Grid 宫格常用布局的实现
2020/01/10 HTML / CSS
html5实现移动端适配完美写法
2017/11/16 HTML / CSS
Speedo速比涛中国官方网站:全球领先泳装运动品牌
2018/04/24 全球购物
大学生村官工作感言
2014/01/10 职场文书
浪漫婚礼主持词
2014/03/14 职场文书
2014离婚协议书范文两篇
2014/09/15 职场文书
长城导游词400字
2015/01/30 职场文书
个人年终总结范文
2015/03/09 职场文书
七年级作文之环保作文
2019/10/17 职场文书
MySQL令人咋舌的隐式转换
2021/04/05 MySQL