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

Public Member Functions

def __init__ (self, weights, num_samples, replacement=True)
 
def __iter__ (self)
 
def __len__ (self)
 
- Public Member Functions inherited from torch.utils.data.sampler.Sampler
def __init__ (self, data_source)
 
def __iter__ (self)
 
def __len__ (self)
 

Public Attributes

 weights
 
 num_samples
 
 replacement
 

Detailed Description

Definition at line 100 of file sampler.py.


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