首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >[tensorflow][原创]tensorflow的ckpt查看节点信息传递路径的问题

[tensorflow][原创]tensorflow的ckpt查看节点信息传递路径的问题

作者头像
云未归来
发布2025-07-18 13:31:34
发布2025-07-18 13:31:34
650
举报

在已有tensorflow的ckpt模型时候,如何查看节点信息,网上有很多代码:

代码语言:javascript
复制
from tensorflow.python import pywrap_tensorflow
checkpoint_path = '路径'
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)

上面没什么问题但是总是报错:

tensorflow.python.framework.errors_impl.NotFoundError: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for ./cnn_emotion_classifier.ckpt/cnn_emotion_classifier.ckpt.data-0000-of-00001

查了一圈网上都没说清楚,其实是路径问题,这个路径不是文件路径也不是文件夹路径,先看的模型路径情况

我试了文件夹里面四个文件绝对路径填进去全部报错,原来应该是文件夹路径+模型名字+.ckpt这样表示,例如上面应该写成

D:/test/model/cnn_emotion_classifier.ckpt

但是这个cnn_emotion_classifier.ckpt并不是某个文件夹和文件名字,虽然这个是低级错误,但是对初学者却不容易解决。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2025-07-17,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档