braindecode.augmentation.AugmentedDataLoader#

class braindecode.augmentation.AugmentedDataLoader(dataset, transforms=None, device=None, **kwargs)[source]#

A base dataloader class customized to applying augmentation Transforms.

Parameters

dataset (BaseDataset) – The dataset containing the signals.
transforms (list | Transform, optional) – Transform or sequence of Transform to be applied to each batch.
device (str | torch.device | None, optional) – Device on which to transform the data. Defaults to None.
**kwargs (dict, optional) – keyword arguments to pass to standard DataLoader class.

Examples using `braindecode.augmentation.AugmentedDataLoader`#