models.rst 2.16 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
Tacotron2
~~~~~~~~~

moto's avatar
moto committed
31
32
33
34
35
36
Model
-----

Tacotoron2
^^^^^^^^^^

37
38
39
40
41
42
43
44
45
46
.. autoclass:: Tacotron2

  .. automethod:: forward

  .. automethod:: infer

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

tacotron2
moto's avatar
moto committed
47
^^^^^^^^^
48
49
50
51

.. autofunction:: tacotron2


moto's avatar
moto committed
52
53
Wav2Letter
~~~~~~~~~~
Tomás Osório's avatar
Tomás Osório committed
54
55
56
57

.. autoclass:: Wav2Letter

  .. automethod:: forward
jimchen90's avatar
jimchen90 committed
58
59


moto's avatar
moto committed
60
61
Wav2Vec2.0 / HuBERT
~~~~~~~~~~~~~~~~~~~
moto's avatar
moto committed
62

moto's avatar
moto committed
63
64
65
Model
-----

moto's avatar
moto committed
66
Wav2Vec2Model
moto's avatar
moto committed
67
^^^^^^^^^^^^^
moto's avatar
moto committed
68
69
70
71
72
73
74
75
76
77

.. autoclass:: Wav2Vec2Model

  .. automethod:: extract_features

  .. automethod:: forward

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

moto's avatar
moto committed
78
wav2vec2_base
moto's avatar
moto committed
79
^^^^^^^^^^^^^
moto's avatar
moto committed
80

moto's avatar
moto committed
81
82
.. autofunction:: wav2vec2_base

moto's avatar
moto committed
83
wav2vec2_large
moto's avatar
moto committed
84
^^^^^^^^^^^^^^
moto's avatar
moto committed
85

moto's avatar
moto committed
86
87
.. autofunction:: wav2vec2_large

moto's avatar
moto committed
88
wav2vec2_large_lv60k
moto's avatar
moto committed
89
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
90

moto's avatar
moto committed
91
92
.. autofunction:: wav2vec2_large_lv60k

93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
wav2vec2_asr_base
^^^^^^^^^^^^^^^^^

.. autofunction:: wav2vec2_asr_base

wav2vec2_asr_large
^^^^^^^^^^^^^^^^^^

.. autofunction:: wav2vec2_asr_large

wav2vec2_asr_large_lv60k
^^^^^^^^^^^^^^^^^^^^^^^^

.. autofunction:: wav2vec2_asr_large_lv60k

moto's avatar
moto committed
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
hubert_base
^^^^^^^^^^^

.. autofunction:: hubert_base

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

.. autofunction:: hubert_large

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

.. autofunction:: hubert_xlarge

hubert_asr_large
^^^^^^^^^^^^^^^^

.. autofunction:: hubert_asr_large

hubert_asr_xlarge
^^^^^^^^^^^^^^^^^

.. autofunction:: hubert_asr_xlarge

133
134
135
136
137
.. currentmodule:: torchaudio.models.wav2vec2.utils

Utility Functions
-----------------

moto's avatar
moto committed
138
import_huggingface_model
moto's avatar
moto committed
139
^^^^^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
140

141
142
.. autofunction:: import_huggingface_model

moto's avatar
moto committed
143
import_fairseq_model
moto's avatar
moto committed
144
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
145

moto's avatar
moto committed
146
147
.. autofunction:: import_fairseq_model

148
.. currentmodule:: torchaudio.models
moto's avatar
moto committed
149

moto's avatar
moto committed
150
151
WaveRNN
~~~~~~~
jimchen90's avatar
jimchen90 committed
152

moto's avatar
moto committed
153
154
155
156
157
158
Model
-----

WaveRNN
^^^^^^^

jimchen90's avatar
jimchen90 committed
159
160
161
.. autoclass:: WaveRNN

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

163
164
  .. automethod:: infer

165
166
167
168
Factory Functions
-----------------

wavernn
moto's avatar
moto committed
169
^^^^^^^
170
171
172

.. autofunction:: wavernn

moto's avatar
moto committed
173
174
175
176
References
~~~~~~~~~~

.. footbibliography::