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