查看tensoflow网络节点名称

最新推荐文章于 2020-12-14 10:36:32 发布

weixin_43214624

最新推荐文章于 2020-12-14 10:36:32 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： tensorflow 文章标签： tensorflow 神经网络 freeze_graph output_node_names

本文链接：https://blog.youkuaiyun.com/weixin_43214624/article/details/89882042

tensorflow 专栏收录该内容

4 篇文章

订阅专栏

博客介绍了在使用TensorFlow的freeze_gragh.py时，输入参数 --output_node_name的处理方法。可在原有网络结构代码后添加所需节点，通过print获取对应tensor名称，如scope_final_output_ids/strided_slice等，最后给出了freeze_graph.py的使用示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在使用freeze_gragh.py时，需要输入参数 --output_node_name，可以在原有网络结构的代码后面直接添加需要的节点，然后在代码中print对应的tensor即可，比如：

# my model :
...
with tf.name_scope("scope_output_node_name"):
    final_output_ids = infer_outputs.predicted_ids[:, :, 0]
    final_score = final_state[1][0][0]
    print(final_output_ids)
    print(final_score)

结果显示两个节点的名称为：
scope_final_output_ids/strided_slice
scope_final_output_ids/strided_slice_2

最后，可以使用freeze_graph.py如下：
python3 freeze_graph.py --input_graph=“models/gqa.batch2.pbtxt” --input_checkpoint=“models/batch_model_epoch0_val2.53_tst0.00.ckpt” --output_node_names=“scope_final_output_ids/strided_slice, scope_final_output_ids/strided_slice_2” --output_graph=“gqa.batch.ckpt.pb”