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

更新时间：2020年02月10日 11:24:19 作者：imumu_xi

今天小编就为大家分享一篇tensorflow实现训练变量checkpoint的保存与读取，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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是循环训练的次数，也就是第几次迭代

以下保存的变量文件

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的保存与读取就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

pycharm配置anaconda环境时找不到python.exe的两种解决办法
如果你在Anaconda中创建了虚拟环境,但是无法找到python.exe,可能是因为虚拟环境的Python路径没有添加到系统环境变量中,这篇文章主要给大家介绍了关于pycharm配置anaconda环境时找不到python.exe的两种解决办法,需要的朋友可以参考下
2024-07-07
python使用技巧-标准输入
这篇文章主要介绍了python使用技巧标准输入,标准输入即stdin ，下文围绕python使用技巧标准输入相关资料展开学习内容，具有一的参考价值，需要的小伙伴可以参考一下
2022-02-02
python多线程下信号处理程序示例
这篇文章主要为大家详细介绍了python多线程下信号处理程序示例，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-05-05
python学生管理系统的实现
这篇文章主要为大家详细介绍了python学生管理系统的实现，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2020-04-04
Numpy实现卷积神经网络(CNN)的示例
这篇文章主要介绍了Numpy实现卷积神经网络(CNN)的示例，帮助大家更好的理解和使用Numpy，感兴趣的朋友可以了解下
2020-10-10
Python实现删除当前目录下除当前脚本以外的文件和文件夹实例
这篇文章主要介绍了Python实现删除当前目录下除当前脚本以外的文件和文件夹的方法,涉及Python针对目录及文件的删除技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-07-07
详解Django3中直接添加Websockets方式
这篇文章主要介绍了Django3中直接添加Websockets方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-02-02
Python NumPy教程之索引详解
这篇文章主要为大家详细介绍了Python NumPy中索引的使用方法，文中的示例代码讲解详细，对我们学习Python有一定帮助，需要的可以参考一下
2022-08-08
python dict 字典以及赋值引用的一些实例(详解)
下面小编就为大家带来一篇python dict 字典以及赋值引用的一些实例(详解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-01-01
Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
今天小编就为大家分享一篇Pytorch 定义MyDatasets实现多通道分别输入不同数据方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01

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

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具