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

Public Member Functions

def __init__ (self, dataset, name, indices, batch_size=1)
 
def setup_ex (self, init_net, exit_net)
 
def reset (self, net)
 
def computeoffset (self, net)
 
def sort_and_shuffle (self, net, sort_by_field=None, shuffle_size=1, batch_size=1)
 
def read (self, read_net)
 

Public Attributes

 dataset
 
 cursor
 
 name
 
 indices
 
 batch_size
 
 offsets
 

Detailed Description

Definition at line 57 of file dataset.py.

Constructor & Destructor Documentation

◆ __init__()

def dataset._DatasetRandomReader.__init__ (   self,
  dataset,
  name,
  indices,
  batch_size = 1 
)
Don't call this directly. Instead, use dataset.random_reader()

Definition at line 58 of file dataset.py.


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