tf.FIFOQueue和data_flow_ops.StagingArea有什么区别?

时间:2017-07-15 13:11:28

标签: tensorflow

为什么StaingArea比FIFOQueue的性能更好?原则上的差异在于它们提供类似的API和功能。

1 个答案:

答案 0 :(得分:1)

请参阅https://www.tensorflow.org/performance/performance_models

队列使用Python线程,因此存在一些瓶颈,并且没有很好的CPU到GPU传输故事。 StagingArea解决了这些问题。

最终tf.contrib.data(不是队列)将有更好的API来实现类似StagingArea的性能。与此同时,在易用性和性能之间存在一些权衡。