Caffe2 - Python API
A deep learning, cross platform ML framework
Classes | Functions | Variables
data_workers Namespace Reference

Module caffe2.python.data_workers. More...

Classes

class  DataInputCoordinator
 
class  GlobalCoordinator
 

Functions

def get_worker_ids (num_workers)
 
def init_data_input_workers (net, input_blob_names, fetch_fun, batch_size, num_worker_threads=2, input_source_name="train", max_buffered_batches=800, init_fun=None, external_loggers=None)
 
def fetcher (coordinator, worker_id, fetch_fun, batch_size, input_blob_names)
 
def enqueuer (coordinator)
 

Variables

 log = logging.getLogger("data_workers")
 
int LOG_INT_SECS = 60
 
 global_coordinator = GlobalCoordinator()
 

Detailed Description

Module caffe2.python.data_workers.