tensorflow 实现打印预训练的模型中的变量名和变量值

tensorflow保存的预训练的模型包含3个文件：

model_name.data-00000-of-00001 (保存网络中每个变量值，包括所有的weights、biases、gradients等变量，类似python字典结果，对应每个变量的数值（key，value）。)
model_name.index （是对应模型的索引文件）
model_name.meta (文件保存的是图结构，meta文件是pb（protocol buffer）格式文件，包含变量、op、集合等。)

打印保存在model里的变量名和变量值：

from tensorflow.python import pywrap_tensorflowcheckpoint_path = "./*/model_name"
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
vars = reader.get_variable_to_shape_map()
for v in vars:v_name = vvar_tensor = reader.get_tensor(v)print("tensor name = ",v)print("tensor shape = ",var_tensor.shape)print("tensor value = ", var_tensor)

tensorflow 实现打印预训练的模型中的变量名和变量值相关推荐

Python中变量名与变量值的关系
在Python中变量名与变量值的只是绑定关系如图,变量名只是一个标识符与变量值绑定在一起而已转载于:https://www.cnblogs.com/liuxiaowei/p/7163470.html
TensorFlow中查看checkpoint文件中的变量名和对应值
在加载模型时, 需要知道checkpoint中变量名称,一下代码可以查看TensorFlow中checkpoint文件中的变量名: #!/usr/bin/env python # -*- coding ...
带你了解3类预训练语音模型预测方法
摘要:在调研多篇论文后,笔者发现预训练语音模型的预测方法基本可以分为以下3类:<1>用前文预测当前及后文(自回归)<2> 随机mask 一些帧并预测 <3> 用两侧 ...
BERT（预训练Transformer模型）
目录一.前言二.随机遮挡,进行预测三.两句话是否原文相邻四.两者结合起来五.总结六.参考链接一.前言 Bert在18年提出,19年发表,Bert的目的是为了预训练Transformer模 ...
在c语言中指针变量名前面的*表示其后面的,C语言中的指针
-------------------------------指针--------------------------------- 指针? 内存单元的编号也叫做地址.既然根据内存单元的号或地址就可 ...
新一代人工智能院士高峰论坛-视觉预训练大模型及其在智慧城市中的应用分论坛顺利举办
2021年12月20日,新一代人工智能院士高峰论坛-视觉预训练大模型及其在智慧城市中的应用分论坛在深圳市人才研修院成功举办. 该论坛由鹏城实验室视觉智能研究所主办,邀请了企业界和学术界的技术大咖和资 ...
keras: 用预训练的模型提取特征
keras提供了VGG19在ImageNet上的预训练权重模型文件,其他可用的模型还有VGG16.Xception.ResNet50.InceptionV3 4个. VGG19在keras中的定义: ...
中文预训练ALBERT模型来了：小模型登顶GLUE，Base版模型小10倍、速度快1倍
(图片由AI科技大本营付费下载自视觉中国) 作者 | 徐亮(实在智能算法专家) 来源 | AINLP(ID:nlpjob) 谷歌ALBERT论文刚刚出炉一周,中文预训练ALBERT模型来了,感兴趣的 ...
紫东太初：自动化所开源图-文-音三模态的预训练大模型
近日,中科院自动化所研发并开源了全球首个图文音(视觉-文本-语音)三模态预训练模型,兼具跨模态理解和生成能力,具有了在无监督情况下多任务联合学习.并快速迁移到不同领域数据的强大能力.自动化所构建了具有 ...

tensorflow 实现打印预训练的模型中的变量名和变量值

tensorflow 实现打印预训练的模型中的变量名和变量值相关推荐

最新文章

热门文章