|
def | __init__ (self, fields, name=None, capacity=1, enforce_unique_name=False, num_threads=1) |
|
def | build (self, reader, process=None) |
|
|
| schema |
|
| name |
|
| num_threads |
|
| blobs_queue |
|
| writer |
|
| reader |
|
| exit_step |
|
The class is used to feed data with some process from a reader into a
queue and provider a reader interface for data fetching from the queue.
Definition at line 48 of file record_queue.py.
def caffe2.python.record_queue.RecordQueue.build |
( |
|
self, |
|
|
|
reader, |
|
|
|
process = None |
|
) |
| |
Build the producer_step to feed data from reader into the queue, and
return the reader interface.
Inputs:
reader: read data which will be stored in the queue.
process: preprocess data before enqueue.
Outputs:
reader: reader to fetch the data from the queue.
producer_step: the step insert the data into the queue. Should be
run with comsume_step together.
exit_step: the step to close queue
schema: the schema for the reader.
Definition at line 80 of file record_queue.py.
The documentation for this class was generated from the following file: