holocron.utils.data¶

Batch collate¶

holocron.utils.data.Mixup(num_classes: int, alpha: float = 0.2) → None[source]¶

Implements a batch collate function with MixUp strategy from “mixup: Beyond Empirical Risk Minimization”.

>>> import torch
>>> from torch.utils.data._utils.collate import default_collate
>>> from holocron.utils.data import Mixup
>>> mix = Mixup(num_classes=10, alpha=0.4)
>>> loader = torch.utils.data.DataLoader(dataset, batch_size, collate_fn=lambda b: mix(*default_collate(b)))

Parameters:

num_classes – number of expected classes
alpha – mixup factor