tensorflow 查看网络节点名称的方法

分类: 文章 • 2024-07-20 20:33:28

在将tensorflow 的.ckpt文件转成pb文件和pb文件测试时需要网络节点的输入和输出名称。

本人是经验是将直接将断点直接打到你需要获取节点名称的位置，然后程序运行到断点处就会出现tensor的名称此名称就是节点对应的名称。因此要获取输入和输出节点名只需要将断点打到输入和输出的位置然后查看tensor的名称就可以了。