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://bzwzjz.com/article/csijos.html

其他资讯

Copyright © 2007-2020 广东宝晨空调科技有限公司 All Rights Reserved 粤ICP备2022107769号
友情链接: app网站建设 重庆网站设计 响应式网站建设 网站设计公司 外贸网站设计方案 成都网站设计 网站建设公司 成都网站建设 重庆企业网站建设 成都网站设计 网站建设 广安网站设计 商城网站建设 上市集团网站建设 重庆网站建设 成都网站建设 成都定制网站建设 成都网站设计 专业网站设计 高端网站设计 成都响应式网站建设公司 成都营销网站制作