我使用重新训练代码在我自己的图像上训练初始模型的最后一层。但是在我的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内存吗?有什么方法可以让它更快?
答案 0 :(得分:0)
Tensorflow使用惰性评估;基本上所有图中的节点都是根据需要在sess.run()进行评估。
这就是为什么你注意到sess.run花费了大部分时间,而不是特别是pool3更昂贵
您可以使用Timeline查找详细的时间细分。