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程序中设置HTTP代理
Nov 06 Python
Jupyter notebook远程访问服务器的方法
May 24 Python
python实现单链表中删除倒数第K个节点的方法
Sep 28 Python
python全栈要学什么 python全栈学习路线
Jun 28 Python
Python Request爬取seo.chinaz.com百度权重网站的查询结果过程解析
Aug 13 Python
基于python进行抽样分布描述及实践详解
Sep 02 Python
python 三元运算符使用解析
Sep 16 Python
tensorflow中tf.slice和tf.gather切片函数的使用
Jan 19 Python
Python填充任意颜色,不同算法时间差异分析说明
May 16 Python
Python新手学习函数默认参数设置
Jun 03 Python
python3爬虫中异步协程的用法
Jul 10 Python
Python基于百度AI实现抓取表情包
Jun 27 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
如何使用PHP获取网络上文件
2006/10/09 PHP
PHP中VC6、VC9、TS、NTS版本的区别与用法详解
2013/10/26 PHP
作为程序员必知的16个最佳PHP库
2015/12/09 PHP
Yii2 ActiveRecord多表关联及多表关联搜索的实现
2016/06/30 PHP
基于Jquery的淡入淡出的特效基础练习
2010/12/13 Javascript
高效的jQuery代码编写技巧总结
2017/02/22 Javascript
vue复合组件实现注册表单功能
2017/11/06 Javascript
Vue精简版风格概述
2018/01/30 Javascript
vue项目实现表单登录页保存账号和密码到cookie功能
2018/08/31 Javascript
vue打包通过image-webpack-loader插件对图片压缩优化操作
2020/11/12 Javascript
vue keep-alive的简单总结
2021/01/25 Vue.js
[47:55]Ti4第二日主赛事败者组 NaVi vs EG 1
2014/07/20 DOTA
python实现定时播放mp3
2015/03/29 Python
Python 判断文件或目录是否存在的实例代码
2018/07/19 Python
python+influxdb+shell编写区域网络状况表
2018/07/27 Python
python是否适合网页编程详解
2019/10/04 Python
解决tensorboard多个events文件显示紊乱的问题
2020/02/15 Python
Python计算IV值的示例讲解
2020/02/28 Python
PyQt5+Pycharm安装和配置图文教程详解
2020/03/24 Python
CSS3使用transition实现的鼠标悬停淡入淡出
2015/01/09 HTML / CSS
英国电动工具购买网站:Anglia Tool Centre
2017/04/25 全球购物
美国知名保健品网站:LuckyVitamin(支持中文)
2017/08/09 全球购物
美国最大的船只买卖在线市场:Boat Trader
2018/08/04 全球购物
澳大利亚玩具剧场:Toy Playhouse
2019/03/03 全球购物
可持续未来的时尚基础:Alternative Apparel
2019/05/06 全球购物
希腊香水和化妆品购物网站:Parfimo.gr
2019/10/03 全球购物
美国传奇滑手Paul Rodriguez创办的街头滑板品牌:Primitive Skateboarding
2019/10/29 全球购物
贝佳斯官方网站:Borghese
2020/05/08 全球购物
车间组长岗位职责
2013/12/20 职场文书
大学生的网上创业计划书
2013/12/31 职场文书
公司门卫岗位职责
2014/03/15 职场文书
协议书样本
2014/04/23 职场文书
公司保洁员岗位职责
2015/02/13 职场文书
2016高考寄语或鼓励的话语
2015/12/04 职场文书
导游词之云南省玉龙雪山
2019/12/19 职场文书
粗暴解决CUDA out of memory的问题
2021/05/22 Python