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各节点名称实例-创新互联
文章地址:http://www.jxjierui.cn/article/csijos.html


咨询
建站咨询
