重庆分公司
重庆分公司，新征程启航
为企业提供网站建设、域名注册、服务器等服务
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: ',keycheckpoint_path为自己的模型路径
以上是“tensorflow如何查看ckpt各节点名称实例”这篇文章的所有内容，感谢各位的阅读！希望分享的内容对大家有帮助，更多相关知识，欢迎关注创新互联行业资讯频道！
网站名称：tensorflow如何查看ckpt各节点名称实例-创新互联
本文URL：http://cqcxhl.com/article/csijos.html