models.rst 1.61 KB
Newer Older
Tomás Osório's avatar
Tomás Osório committed
1
2
3
4
.. role:: hidden
    :class: hidden-section

torchaudio.models
moto's avatar
moto committed
5
=================
Tomás Osório's avatar
Tomás Osório committed
6
7
8
9
10
11

.. currentmodule:: torchaudio.models

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


moto's avatar
moto committed
12
13
ConvTasNet
~~~~~~~~~~
moto's avatar
moto committed
14
15
16
17
18
19

.. autoclass:: ConvTasNet

  .. automethod:: forward


moto's avatar
moto committed
20
21
DeepSpeech
~~~~~~~~~~
discort's avatar
discort committed
22
23
24
25
26
27

.. autoclass:: DeepSpeech

  .. automethod:: forward


28
29
30
31
32
33
34
35
36
Tacotron2
~~~~~~~~~

.. autoclass:: Tacotron2

  .. automethod:: forward

  .. automethod:: infer

moto's avatar
moto committed
37
38
Wav2Letter
~~~~~~~~~~
Tomás Osório's avatar
Tomás Osório committed
39
40
41
42

.. autoclass:: Wav2Letter

  .. automethod:: forward
jimchen90's avatar
jimchen90 committed
43
44


moto's avatar
moto committed
45
46
Wav2Vec2.0 / HuBERT
~~~~~~~~~~~~~~~~~~~
moto's avatar
moto committed
47

moto's avatar
moto committed
48
49
50
Model
-----

moto's avatar
moto committed
51
Wav2Vec2Model
moto's avatar
moto committed
52
^^^^^^^^^^^^^
moto's avatar
moto committed
53
54
55
56
57
58
59
60
61
62

.. autoclass:: Wav2Vec2Model

  .. automethod:: extract_features

  .. automethod:: forward

Factory Functions
-----------------

63
64
65
66
67
68
wav2vec2_model
^^^^^^^^^^^^^^

.. autofunction:: wav2vec2_model


moto's avatar
moto committed
69
wav2vec2_base
moto's avatar
moto committed
70
^^^^^^^^^^^^^
moto's avatar
moto committed
71

moto's avatar
moto committed
72
73
.. autofunction:: wav2vec2_base

moto's avatar
moto committed
74
wav2vec2_large
moto's avatar
moto committed
75
^^^^^^^^^^^^^^
moto's avatar
moto committed
76

moto's avatar
moto committed
77
78
.. autofunction:: wav2vec2_large

moto's avatar
moto committed
79
wav2vec2_large_lv60k
moto's avatar
moto committed
80
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
81

moto's avatar
moto committed
82
83
.. autofunction:: wav2vec2_large_lv60k

moto's avatar
moto committed
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
hubert_base
^^^^^^^^^^^

.. autofunction:: hubert_base

hubert_large
^^^^^^^^^^^^

.. autofunction:: hubert_large

hubert_xlarge
^^^^^^^^^^^^^

.. autofunction:: hubert_xlarge

99
100
101
Utility Functions
-----------------

102
103
.. currentmodule:: torchaudio.models.wav2vec2.utils

moto's avatar
moto committed
104
import_huggingface_model
moto's avatar
moto committed
105
^^^^^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
106

107
108
.. autofunction:: import_huggingface_model

moto's avatar
moto committed
109
import_fairseq_model
moto's avatar
moto committed
110
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
111

moto's avatar
moto committed
112
113
.. autofunction:: import_fairseq_model

114
.. currentmodule:: torchaudio.models
moto's avatar
moto committed
115

moto's avatar
moto committed
116
117
WaveRNN
~~~~~~~
jimchen90's avatar
jimchen90 committed
118
119
120
121

.. autoclass:: WaveRNN

  .. automethod:: forward
moto's avatar
moto committed
122

123
124
  .. automethod:: infer

moto's avatar
moto committed
125
126
127
128
References
~~~~~~~~~~

.. footbibliography::