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

Public Member Functions

def __init__ (self, dataset, name, batch_size=1, enforce_batch_size=False)
 
def setup_ex (self, init_net, exit_net)
 
def read (self, read_net)
 
def reset (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)
 

Public Attributes

 dataset
 
 name
 
 batch_size
 
 enforce_batch_size
 
 cursor
 

Detailed Description

Definition at line 25 of file dataset.py.

Constructor & Destructor Documentation

def caffe2.python.dataset._DatasetReader.__init__ (   self,
  dataset,
  name,
  batch_size = 1,
  enforce_batch_size = False 
)
Don't call this directly. Instead, use dataset.reader()

Definition at line 26 of file dataset.py.


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