tensorflow如何查看ckpt各节点名称实例-创新互联

这篇文章主要介绍tensorflow如何查看ckpt各节点名称实例，文中介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们一定要看完！

让客户满意是我们工作的目标，不断超越客户的期望值来自于我们对这个行业的热爱。我们立志把好的技术通过有效、简单的方式提供给客户，将通过不懈努力成为客户在信息化领域值得信任、有价值的长期合作伙伴，公司提供的服务项目有：域名注册、雅安服务器托管、营销软件、网站建设、拜泉网站维护、网站推广。

运行下列脚本，可以打印出模型各个节点变量的名称：

from tensorflow.python import pywrap_tensorflow
import os

checkpoint_path=os.path.join('model.ckpt-131805')
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

checkpoint_path为自己的模型路径

以上是“tensorflow如何查看ckpt各节点名称实例”这篇文章的所有内容，感谢各位的阅读！希望分享的内容对大家有帮助，更多相关知识，欢迎关注创新互联行业资讯频道！

网站名称：tensorflow如何查看ckpt各节点名称实例-创新互联
转载注明：http://cdiso.cn/article/csijos.html