from tensorflow.python import pywrap_tensorflow
import os
model_dir="./"
checkpoint_path = os.path.join(model_dir, "model.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
print(reader.get_tensor(key))
输出checkpoint变量名及其值
TensorFlow模型检查
最新推荐文章于 2022-12-29 11:47:23 发布
本文介绍了一个使用Python脚本读取TensorFlow模型文件的方法,并展示了如何列出模型中的所有变量及其形状。此过程对于理解模型结构及参数配置非常有用。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
1万+

被折叠的 条评论
为什么被折叠?



