Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.text_file_reader.TextFileReader Class Reference
Inheritance diagram for caffe2.python.text_file_reader.TextFileReader:
caffe2.python.dataio.Reader

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)
 

Detailed Description

Wrapper around operators for reading from text files.

Definition at line 12 of file text_file_reader.py.

Constructor & Destructor Documentation

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.

Member Function Documentation

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.


The documentation for this class was generated from the following file: