RandomSampler#

class RandomSampler(dataset, batch_size=1, drop_last=False, indices=None, world_size=None, rank=None, seed=None)[source]#

Sample elements randomly without replacement.

Parameters:

  • dataset – dataset to sample from.

  • batch_size – batch size for batch method.

  • drop_last – set True to drop the last incomplete batch, if the dataset size is not divisible by the batch size. If False and the size of dataset is not divisible by the batch_size, then the last batch will be smaller. Default: False

  • indices – indice of samples.

  • world_size – number of ranks.

  • rank – rank id, non-negative interger within 0 and world_size.

  • seed – seed for random operators.

sample()[source]#

Return a list contains all sample indices.

Return type:

List