ckpt转npy、转csv、转pickle详解

本文介绍了如何将TensorFlow的checkpoint文件转换为npy、csv和pickle格式。通过示例代码,展示了如何读取checkpoint权重,将其保存为dict并存储在.npy文件中,然后由于.csv保存时会导致数据变为字符串类型,所以选择了pickle库来正确保存和读取权重值。
摘要由CSDN通过智能技术生成

因为程序里的参数需要直接加载训练好的权重值,在此基础上再进行训练,所以需要在另一个程序里训练好权重,先将其保存到本地,然后该程序再来加载使用。

我们知道,通过以下代码,在训练过程中会得到checkpoint文件,里面会存储着训练中所有训练过的变量的具体数值:

saver = tf.train.Saver()

checkpoint_name = os.path.join(checkpoint_path, 'model_epoch'+str(epoch+1)+'.ckpt')

save_path = saver.save(sess, checkpoint_name)

得到的本地文件如下图所示,每三个这样的为一组:


那么我们如何查看checkpoint文件,也就是上述这几个文件里保存下来的权重值呢?具体代码见以下链接。很详细:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py

按照我在文章开头所说,我希望的是把文件里面的权重值给保存在本地,而不是这样一个抽象的文件,以后加载啥的也不方便,于是我利用了上述链接里的几个小函数&#

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值