Tensorflow缓慢地从一个再培训模型中提取pool3特征

Question

我使用培训代码来训练我自己的图像上的Inception模型的最后一层。但在我的CPU机器上，它需要将近0.5-0.8秒才能提取pool3功能。

pool3 = sess.graph.get_tensor_by_name('pool_3:0') 
pf = sess.run(pool3, {png_data: imgBuffer.getvalue()}) 

上述代码行占用了所有时间。是否因为我的CPU内存？任何方式使其更快？

Answer 1

Tensorflow使用懒惰评估;基本上图中的所有节点都是在sess.run（）的需求基础上进行评估的。

这就是为什么你注意到大部分时间由sess.run度过，尤其是

您可以使用Timeline找到详细的时间分解更昂贵的它不是pool3。