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

Public Member Functions

def __init__ (self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=default_collate, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None)
 
def __setattr__ (self, attr, val)
 
def __iter__ (self)
 
def __len__ (self)
 

Public Attributes

 dataset
 
 batch_size
 
 num_workers
 
 collate_fn
 
 pin_memory
 
 drop_last
 
 timeout
 
 worker_init_fn
 
 sampler
 
 batch_sampler
 

Detailed Description

Definition at line 24 of file dataloader.py.


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