Public Member Functions | |
def | __init__ (self, init_net, filename, schema, num_passes=1, batch_size=1) |
def | read (self, net) |
Public Member Functions inherited from caffe2.python.dataio.Reader | |
def | __init__ (self, schema=None) |
def | schema (self) |
def | setup_ex (self, init_net, finish_net) |
def | read_ex (self, local_init_net, local_finish_net) |
def | read_record_ex (self, local_init_net, local_finish_net) |
def | read (self, read_net) |
def | reset (self, net) |
def | read_record (self, read_net) |
def | execution_step (self, reader_net_name=None, external_should_stop=None) |
Wrapper around operators for reading from text files.
Definition at line 12 of file text_file_reader.py.
def caffe2.python.text_file_reader.TextFileReader.__init__ | ( | self, | |
init_net, | |||
filename, | |||
schema, | |||
num_passes = 1 , |
|||
batch_size = 1 |
|||
) |
Create op for building a TextFileReader instance in the workspace. Args: init_net : Net that will be run only once at startup. filename : Path to file to read from. schema : schema.Struct representing the schema of the data. Currently, only support Struct of strings. num_passes : Number of passes over the data. batch_size : Number of rows to read at a time.
Definition at line 16 of file text_file_reader.py.
def caffe2.python.text_file_reader.TextFileReader.read | ( | self, | |
net | |||
) |
Create op for reading a batch of rows.
Definition at line 42 of file text_file_reader.py.