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#

Searching the best data augmentation on BCIC IV 2a Dataset

Searching the best data augmentation on BCIC IV 2a Dataset