augment_dataloader

maite.tasks.augment_dataloader(*, augmentation, dataloader)

Create an DataLoader of augmented inputs from a Dataset or DataLoader.

Parameters:

augmentationAugmentation

Compatible maite augmentation.

dataloaderDataloader

Compatible maite dataloader.

Returns:

DataLoader[InputType, TargetType, DatumMetadataType]

DataLoader for augmented inputs.