cerebras.modelzoo.data.vision.classification.sampler.RepeatedAugSampler#

class cerebras.modelzoo.data.vision.classification.sampler.RepeatedAugSampler(*args, **kwargs)[source]#

Bases: torch.utils.data.Sampler

Sampler that restricts data loading to a subset of the dataset for distributed, with repeated augmentation. It ensures that different each augmented version of a sample will be visible to a different process (GPU). Heavily based on ‘torch.utils.data.DistributedSampler’.

This is borrowed from the DeiT Repo: https://github.com/facebookresearch/deit/blob/main/samplers.py

Methods

set_epoch