Tensorflowがリトレーニングモデルからpool3フィーチャーを抽出するのが遅い

Question

リトレインコードを使用して、自分自身のイメージでインセプションモデルの最終レイヤーをトレーニングしました。しかし、私のCPUマシンでは、pool3の機能を抽出するのにほぼ0.5-0.8秒かかります。

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によって費やされていると、それは特にそのpool3ではありませんが、より高価な

であるあなたは、詳細な時間の内訳を見つけるためにTimelineを使用することができます。