MixUpDataset

class torchelie.datasets.MixUpDataset(dataset, alpha=0.4, transform=None)

Linearly mixes two samples and labels from a dataset according to the MixUp algorithm

https://arxiv.org/abs/1905.02249

Parameters

  • dataset (Dataset) – the dataset

  • alpha (float) – the alpha that parameterizes the beta distribution from which the blending factor is sampled