如何使用张量板找到输出节点?

时间:2019-02-09 05:49:00

标签: tensorflow

我设法训练了模型并得到了检查点。 https://github.com/BichenWuUCB/squeezeDet

在尝试冻结图形时,我需要知道输出节点。

我不明白的是,节点到底是什么?

问题: 1.如何从张量板中找出什么是节点? 2.我正在尝试进行推理的一个更基本的问题,freezegraph是正确的方法吗?

谢谢。 MK

1 个答案:

答案 0 :(得分:0)

节点是张量。因此,输出节点是输出张量,即具有您想要获取的值的张量。

在tensorflow中,您构建了计算图,这就是为什么它被称为节点的原因。

是的,冻结图的推理部分是部署推理的好方法。