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

作者:imumu_xi 时间:2023-12-15 18:10:33 

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 存放的是你新网络中所有的变量

来源:https://blog.csdn.net/sinat_30372583/article/details/79763044

标签:tensorflow,checkpoint,保存,读取
0
投稿

猜你喜欢

  • 超详细注释之OpenCV操作图像平移转换

    2022-08-14 19:29:38
  • pytorch如何利用ResNet18进行手写数字识别

    2022-02-07 04:07:26
  • 基于Python的ModbusTCP客户端实现详解

    2022-02-03 10:54:25
  • python 用正则表达式筛选文本信息的实例

    2023-04-29 14:12:26
  • CentOS 7上为PHP5安装suPHP的方法(彭哥)

    2024-03-24 23:34:12
  • django中的图片验证码功能

    2022-06-10 00:07:54
  • Go并发编程之sync.Once使用实例详解

    2024-04-26 17:21:23
  • python判断列表的连续数字范围并分块的方法

    2021-01-18 12:04:19
  • Echarts实例教程之树形图表的实现方法

    2024-04-18 09:44:01
  • js实现GIF动图分解成多帧图片上传

    2023-08-15 00:06:38
  • 在vue项目中使用Jquery-contextmenu插件的步骤讲解

    2023-07-02 17:08:29
  • python 边缘扩充方式的实现示例

    2021-03-30 23:28:17
  • 设置密码保护的SqlServer数据库备份文件与恢复文件的方法

    2024-01-15 10:19:55
  • MySQL数据库中删除重复记录的方法总结[推荐]

    2024-01-13 07:13:45
  • Linux下如何实现Mysql定时任务

    2024-01-19 09:41:50
  • python 使用Tensorflow训练BP神经网络实现鸢尾花分类

    2023-04-15 13:29:00
  • Python利用DNN实现宝石识别

    2023-08-07 05:49:18
  • SQL Server 2008 安装SQLDMO.dll的方法

    2024-01-12 18:13:15
  • Golang中 import cycle not allowed 问题的解决方法

    2024-04-25 15:13:38
  • 30步检查SQL Server安全列表

    2008-12-18 14:28:00
  • asp之家 网络编程 m.aspxhome.com