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。