pipelines.rst 4.89 KB
Newer Older
1
2
3
4
5
torchaudio.pipelines
====================

.. currentmodule:: torchaudio.pipelines

moto's avatar
moto committed
6
7
.. py:module:: torchaudio.pipelines
		   
8
9
10
11
12
13
The pipelines subpackage contains API to access the models with pretrained weights, and information/helper functions associated the pretrained weights.

wav2vec 2.0 / HuBERT - Representation Learning
----------------------------------------------

.. autoclass:: Wav2Vec2Bundle
14
   :members: sample_rate
15
16
17
18

   .. automethod:: get_model

WAV2VEC2_BASE
19
~~~~~~~~~~~~~
20
21
22
23
24
25
26

.. container:: py attribute

   .. autodata:: WAV2VEC2_BASE
      :no-value:

WAV2VEC2_LARGE
27
~~~~~~~~~~~~~~
28
29
30
31
32
33
34

.. container:: py attribute

   .. autodata:: WAV2VEC2_LARGE
      :no-value:

WAV2VEC2_LARGE_LV60K
35
~~~~~~~~~~~~~~~~~~~~
36
37
38
39
40
41
42
43

.. container:: py attribute

   .. autodata:: WAV2VEC2_LARGE_LV60K
      :no-value:


WAV2VEC2_XLSR53
44
~~~~~~~~~~~~~~~
45
46
47
48
49
50
51

.. container:: py attribute

   .. autodata:: WAV2VEC2_XLSR53
      :no-value:

HUBERT_BASE
52
~~~~~~~~~~~
53
54
55
56
57
58
59

.. container:: py attribute

   .. autodata:: HUBERT_BASE
      :no-value:

HUBERT_LARGE
60
~~~~~~~~~~~~
61
62
63
64
65
66
67

.. container:: py attribute

   .. autodata:: HUBERT_LARGE
      :no-value:

HUBERT_XLARGE
68
~~~~~~~~~~~~~
69
70
71
72
73
74

.. container:: py attribute

   .. autodata:: HUBERT_XLARGE
      :no-value:

75
76
wav2vec 2.0 / HuBERT - Fine-tuned ASR
-------------------------------------
77

moto's avatar
moto committed
78
79
80
Wav2Vec2ASRBundle
~~~~~~~~~~~~~~~~~

81
.. autoclass:: Wav2Vec2ASRBundle
82
   :members: sample_rate
83
84
85
86
87
88

   .. automethod:: get_model

   .. automethod:: get_labels

WAV2VEC2_ASR_BASE_10M
89
~~~~~~~~~~~~~~~~~~~~~
90
91
92
93
94
95
96

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_BASE_10M
      :no-value:

WAV2VEC2_ASR_BASE_100H
97
~~~~~~~~~~~~~~~~~~~~~~
98
99
100
101
102
103
104
      
.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_BASE_100H
      :no-value:

WAV2VEC2_ASR_BASE_960H
105
~~~~~~~~~~~~~~~~~~~~~~
106
107
108
109
110
111
112

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_BASE_960H
      :no-value:

WAV2VEC2_ASR_LARGE_10M
113
~~~~~~~~~~~~~~~~~~~~~~
114
115
116
117
118
119
120

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_10M
      :no-value:

WAV2VEC2_ASR_LARGE_100H
121
~~~~~~~~~~~~~~~~~~~~~~~
122
123
124
125
126
127
128

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_100H
      :no-value:

WAV2VEC2_ASR_LARGE_960H
129
~~~~~~~~~~~~~~~~~~~~~~~
130
131
132
133
134
135
136

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_960H
      :no-value:

WAV2VEC2_ASR_LARGE_LV60K_10M
137
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
138
139
140
141
142
143
144

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_10M
      :no-value:

WAV2VEC2_ASR_LARGE_LV60K_100H
145
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
146
147
148
149
150
151
152

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_100H
      :no-value:

WAV2VEC2_ASR_LARGE_LV60K_960H
153
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
154
155
156
157
158
159

.. container:: py attribute

   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_960H
      :no-value:

160
161
162
163
164
165
166
167
VOXPOPULI_ASR_BASE_10K_DE
~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: VOXPOPULI_ASR_BASE_10K_DE
      :no-value:

168
169
170
171
172
173
174
175
VOXPOPULI_ASR_BASE_10K_EN
~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: VOXPOPULI_ASR_BASE_10K_EN
      :no-value:

176
177
178
179
180
181
182
183
VOXPOPULI_ASR_BASE_10K_ES
~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: VOXPOPULI_ASR_BASE_10K_ES
      :no-value:

184
185
186
187
188
189
190
191
VOXPOPULI_ASR_BASE_10K_FR
~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: VOXPOPULI_ASR_BASE_10K_FR
      :no-value:

192
193
194
195
196
197
198
199
VOXPOPULI_ASR_BASE_10K_IT
~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: VOXPOPULI_ASR_BASE_10K_IT
      :no-value:

200
HUBERT_ASR_LARGE
201
~~~~~~~~~~~~~~~~
202
203
204
205
206
207
208

.. container:: py attribute

   .. autodata:: HUBERT_ASR_LARGE
      :no-value:

HUBERT_ASR_XLARGE
209
~~~~~~~~~~~~~~~~~
210
211
212
213

.. container:: py attribute

   .. autodata:: HUBERT_ASR_XLARGE
moto's avatar
moto committed
214
      :no-value:
moto's avatar
moto committed
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276

Tacotron2 Text-To-Speech
------------------------

Tacotron2TTSBundle
~~~~~~~~~~~~~~~~~~

.. autoclass:: Tacotron2TTSBundle

   .. automethod:: get_text_processor

   .. automethod:: get_tacotron2

   .. automethod:: get_vocoder

Tacotron2TTSBundle - TextProcessor
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: torchaudio.pipelines::Tacotron2TTSBundle.TextProcessor
   :members: tokens
   :special-members: __call__


Tacotron2TTSBundle - Vocoder
~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: torchaudio.pipelines::Tacotron2TTSBundle.Vocoder
   :members: sample_rate
   :special-members: __call__


TACOTRON2_WAVERNN_PHONE_LJSPEECH
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: TACOTRON2_WAVERNN_PHONE_LJSPEECH
      :no-value:


TACOTRON2_WAVERNN_CHAR_LJSPEECH
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: TACOTRON2_WAVERNN_CHAR_LJSPEECH
      :no-value:

TACOTRON2_GRIFFINLIM_PHONE_LJSPEECH
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: TACOTRON2_GRIFFINLIM_PHONE_LJSPEECH
      :no-value:

TACOTRON2_GRIFFINLIM_CHAR_LJSPEECH
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. container:: py attribute

   .. autodata:: TACOTRON2_GRIFFINLIM_CHAR_LJSPEECH
277
278
279
280
281
282
      :no-value:

References
----------

.. footbibliography::