double_buffer

paddle.fluid.layers.io. double_buffer ( reader, place=None, name=None ) [source]

Wrap a double buffer reader. The class Reader contains DecoratedReader and FileReader. Moreover, the DecoratedReader is inherited by CustomReader and BufferedReader. This function is related to BufferedReader. The data will copy to target place with a double buffer queue. If the target place is None, the place that executor perform on will be used.

Parameters
  • reader (Variable) – The Reader Variable need to be wrapped.

  • place (Place|str, optional) – The place of target data, such as CPU, GPU, and if use GPU, it’s necessary to point out which card is involved. Default is the sample place of executor perform. if place is string, It can be cpu, gpu:x, where x is the ndex of the GPUs.

  • name (str, optional) – Variable name. Normally there is no need for user to set this property. For more information, please refer to Name. Default is None.

Returns

wrapped reader with double buffer.

Return type

Variable(Reader)

Examples

import paddle.fluid as fluid
reader = fluid.layers.py_reader(capacity=64,
                                shapes=[(-1, 1, 28, 28), (-1, 1)],
                                dtypes=['float32', 'int64'],
                                use_double_buffer=False)
reader = fluid.layers.double_buffer(reader)
image, label = fluid.layers.read_file(reader)