cerebras.modelzoo.data.vision.classification.sampler

Classes

RepeatedAugSampler

Sampler that restricts data loading to a subset of the dataset for distributed, with repeated augmentation.

previous

cerebras.modelzoo.data.vision.classification.mixup.RandomMixup

next

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