torchaudio.prototype.models

Warning

Starting with version 2.8, we are refactoring TorchAudio to transition it into a maintenance phase. As a result, the prototype module is deprecated in 2.8 and will be removed in 2.9.

The torchaudio.prototype.models subpackage contains definitions of models for addressing common audio tasks.

Note

For models with pre-trained parameters, please refer to torchaudio.prototype.pipelines module.

Model defintions are responsible for constructing computation graphs and executing them.

Some models have complex structure and variations. For such models, factory functions are provided.

ConformerWav2Vec2PretrainModel	DEPRECATED
ConvEmformer	DEPRECATED
HiFiGANVocoder	DEPRECATED

Prototype Factory Functions of Beta Models

Some model definitions are in beta, but there are new factory functions that are still in prototype. Please check “Prototype Factory Functions” section in each model.

Wav2Vec2Model	Acoustic model used in wav2vec 2.0 [Baevski et al., 2020].
RNNT	Recurrent neural network transducer (RNN-T) model.