1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
.. py:module:: torchaudio.models
torchaudio.models
=================
.. currentmodule:: torchaudio.models
The ``torchaudio.models`` subpackage contains definitions of models for addressing common audio tasks.
.. note::
For models with pre-trained parameters, please refer to :mod:`torchaudio.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.
.. autosummary::
:toctree: generated
:nosignatures:
:template: autosummary/model_class.rst
Conformer
ConvTasNet
DeepSpeech
Emformer
HDemucs
HuBERTPretrainModel
RNNT
RNNTBeamSearch
SquimObjective
SquimSubjective
Tacotron2
Wav2Letter
Wav2Vec2Model
WaveRNN
|