我设法训练了模型并得到了检查点。 https://github.com/BichenWuUCB/squeezeDet
在尝试冻结图形时,我需要知道输出节点。
我不明白的是,节点到底是什么?
问题: 1.如何从张量板中找出什么是节点? 2.我正在尝试进行推理的一个更基本的问题,freezegraph是正确的方法吗?
谢谢。 MK
答案 0 :(得分:0)
节点是张量。因此,输出节点是输出张量,即具有您想要获取的值的张量。
在tensorflow中,您构建了计算图,这就是为什么它被称为节点的原因。
是的,冻结图的推理部分是部署推理的好方法。