torch_ecg.augmenters.AUGMENTERS

torch_ecg.augmenters.AUGMENTERS = Registry(name=augmenters, items=['BaselineWanderAugmenter', 'baseline_wander', 'CutMix', 'cutmix', 'LabelSmooth', 'label_smooth', 'Mixup', 'mixup', 'RandomFlip', 'random_flip', 'RandomMasking', 'random_masking', 'RandomRenormalize', 'random_renormalize', 'StretchCompress', 'stretch_compress'])

Registry for managing and building modules.

A registry is used to map strings (module names) to classes, and provides a unified interface to instantiate modules from configurations.

Parameters:

name (str) – Name of the registry.

Examples

>>> BACKBONES = Registry("backbones")
>>> @BACKBONES.register()
... class ResNet(nn.Module):
...     def __init__(self, depth):
...         self.depth = depth
>>> # Build from string
>>> model = BACKBONES.build("ResNet", depth=50)
>>> # Build from config dict
>>> model = BACKBONES.build({"name": "ResNet", "depth": 101})