ertk.preprocessing.speechbrain.SpeechBrainExtractorConfig

class ertk.preprocessing.speechbrain.SpeechBrainExtractorConfig(model: str = '???', device: str = 'cuda', task: Task = Task.EMBEDDINGS, agg: Agg | None = Agg.MEAN, max_input_len: int = 1500000, normalise: bool = True)

Bases: ERTKConfig

SpeechBrain feature extractor configuration.

__init__(model: str = '???', device: str = 'cuda', task: Task = Task.EMBEDDINGS, agg: Agg | None = Agg.MEAN, max_input_len: int = 1500000, normalise: bool = True) → None

Methods

__init__([model, device, task, agg, ...])

Inherited Methods

default()	Create default config.
from_config(config)	Create config object from any compatible config.
from_file(path)	Create config from YAML file and optionlly override some values.
merge_with_args(args)	Merge config with command-line arguments.
merge_with_config(config)	Merge other config into this config.
to_dictconfig()	Convert config to DictConfig.
to_file(path)	Write config to YAML file.
to_string()	Generate YAML string representation of config.

Attributes

agg	Aggregation method for embeddings.
device	Device to run model on.
max_input_len	Maximum input length.
model	Model name or path.
normalise	Normalise audio.
task	Task to perform.

agg: Agg | None = 'mean'

Aggregation method for embeddings.

device: str = 'cuda'

Device to run model on.

max_input_len: int = 1500000

Maximum input length.

model: str = '???'

Model name or path.

normalise: bool = True

Normalise audio.

task: Task = 'embeddings'

Task to perform.