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是循环训练的次数,也就是第几次迭代
以下保存的变量文件
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
投稿
猜你喜欢
Python MD5文件生成码
2023-01-21 02:41:31
HTML 标签是否匹配检测代码
2010-03-17 20:50:00
javascript面向对象编程(四)
2008-03-07 14:20:00
Python时间获取及转换知识汇总
2023-08-02 12:17:08
Python Flask请求扩展与中间件相关知识总结
2022-03-10 17:03:06
浅谈MySQL的存储引擎之表类型存储
2011-05-05 08:10:00
用Dreamweaver MX制作导航下拉菜单
2009-05-29 18:37:00
CSS的渲染效率:书写高效的CSS
2008-11-13 16:55:00
Python使用matplotlib简单绘图示例
2023-11-03 01:33:00
如何在事件代理中正确使用 focus 和 blur 事件
2010-01-30 12:51:00
再谈float菜单局中
2009-12-21 19:57:00
MYSQL教程:检查数据表和修复数据表
2009-03-11 15:24:00
python集合比较(交集,并集,差集)方法详解
2023-09-30 22:11:13
Python的函数嵌套的使用方法
2022-03-23 10:28:54
eWebEditor不支持IE8的解决方法
2009-11-02 10:59:00
在MySQL数据库中如何来复位根用户的密码
2008-12-03 16:57:00
Python实现输出某区间范围内全部素数的方法
2022-04-25 10:04:50
关于Python排序sort()函数和sorted()函数
2023-10-31 01:32:14
Python方法的延迟加载的示例代码
2022-09-05 03:25:00
FSO组件之驱动器操作asp源码
2010-05-03 10:59:00