热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

将tensorflow检查点保存到.pbprotobuf文件

如何解决《将tensorflow检查点保存到.pbprotobuf文件》经验,谁能帮忙解答一下?

我已经在tensorflow上训练了一个pix2pix模型,并且该模型已经以检查点的形式保存了以下文件:

model-15000.metamodel-15000.indexmodel-15000.data-00000-of-00001graph.pbtxtcheckpoint

现在,我想将其转换为protobuf文件(.pb)以进行部署。我遇到了freeze_graph.py脚本来执行此操作,但是我遇到了其中一个参数的麻烦,它是output_node_names

我已经尝试了几个图层名称,但是出现以下错误:

断言错误:生成器/解码器_2 / batchnorm /比例/渐变不在图中

不确定如何找到output_node_names


推荐阅读
author-avatar
笃志单车小博_801
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有