Skip to content

Commit b2f91c2

Browse files
nateanlfacebook-github-bot
authored andcommitted
Import torchaudio #1890 211270d
Reviewed By: mthrok Differential Revision: D31728916 fbshipit-source-id: 4b932f285c274c5f2197325ce73ecbd930e3597e
1 parent 1910dbc commit b2f91c2

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

50 files changed

+2258
-1797
lines changed

.circleci/config.yml

Lines changed: 101 additions & 417 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

.circleci/config.yml.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -517,7 +517,7 @@ jobs:
517517
name: windows-gpu
518518
environment:
519519
<<: *environment
520-
CUDA_VERSION: "10.2"
520+
CUDA_VERSION: "11.3"
521521
TORCHAUDIO_TEST_FORCE_CUDA: 1
522522
steps:
523523
- checkout

.circleci/regenerate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"]
2424
CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111","cu113", "rocm4.1"],
25-
"windows": ["cpu", "cu102", "cu111"],
25+
"windows": ["cpu", "cu113"],
2626
"macos": ["cpu"]}
2727

2828

@@ -209,7 +209,7 @@ def unittest_workflows(indentation=6):
209209
job = {
210210
"name": f"unittest_{os_type}_{device_type}_py{python_version}",
211211
"python_version": python_version,
212-
"cuda_version": 'cpu' if device_type=="cpu" else "cu102",
212+
"cuda_version": 'cpu' if device_type == "cpu" else "cu113",
213213
}
214214

215215
if os_type != "windows":
@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6):
222222
"stylecheck": {
223223
"name": f"stylecheck_py{python_version}",
224224
"python_version": python_version,
225-
"cuda_version": 'cpu' if device_type=="cpu" else "cu102",
225+
"cuda_version": "cpu",
226226
}
227227
})
228228
return indent(indentation, jobs)

docs/source/index.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ The :mod:`torchaudio` package consists of I/O, popular datasets and common audio
3535
transforms
3636
datasets
3737
models
38+
pipelines
3839
sox_effects
3940
compliance.kaldi
4041
kaldi_io

docs/source/models.rst

Lines changed: 0 additions & 208 deletions
Original file line numberDiff line numberDiff line change
@@ -28,27 +28,12 @@ DeepSpeech
2828
Tacotron2
2929
~~~~~~~~~
3030

31-
Model
32-
-----
33-
34-
Tacotoron2
35-
^^^^^^^^^^
36-
3731
.. autoclass:: Tacotron2
3832

3933
.. automethod:: forward
4034

4135
.. automethod:: infer
4236

43-
Factory Functions
44-
-----------------
45-
46-
tacotron2
47-
^^^^^^^^^
48-
49-
.. autofunction:: tacotron2
50-
51-
5237
Wav2Letter
5338
~~~~~~~~~~
5439

@@ -96,21 +81,6 @@ wav2vec2_large_lv60k
9681

9782
.. autofunction:: wav2vec2_large_lv60k
9883

99-
wav2vec2_ft_base
100-
^^^^^^^^^^^^^^^^^
101-
102-
.. autofunction:: wav2vec2_ft_base
103-
104-
wav2vec2_ft_large
105-
^^^^^^^^^^^^^^^^^^
106-
107-
.. autofunction:: wav2vec2_ft_large
108-
109-
wav2vec2_ft_large_lv60k
110-
^^^^^^^^^^^^^^^^^^^^^^^^
111-
112-
.. autofunction:: wav2vec2_ft_large_lv60k
113-
11484
hubert_base
11585
^^^^^^^^^^^
11686

@@ -126,170 +96,6 @@ hubert_xlarge
12696

12797
.. autofunction:: hubert_xlarge
12898

129-
hubert_ft_large
130-
^^^^^^^^^^^^^^^^
131-
132-
.. autofunction:: hubert_ft_large
133-
134-
hubert_ft_xlarge
135-
^^^^^^^^^^^^^^^^^
136-
137-
.. autofunction:: hubert_ft_xlarge
138-
139-
Pre-trained Models
140-
------------------
141-
142-
.. autoclass:: Wav2Vec2PretrainedModelBundle
143-
144-
.. automethod:: get_model
145-
146-
.. autoproperty:: labels
147-
148-
149-
WAV2VEC2_BASE
150-
^^^^^^^^^^^^^
151-
152-
.. container:: py attribute
153-
154-
.. autodata:: WAV2VEC2_BASE
155-
:no-value:
156-
157-
WAV2VEC2_ASR_BASE_10M
158-
^^^^^^^^^^^^^^^^^^^^^
159-
160-
.. container:: py attribute
161-
162-
.. autodata:: torchaudio.models.WAV2VEC2_ASR_BASE_10M
163-
:no-value:
164-
165-
WAV2VEC2_ASR_BASE_100H
166-
^^^^^^^^^^^^^^^^^^^^^^
167-
168-
.. container:: py attribute
169-
170-
.. autodata:: WAV2VEC2_ASR_BASE_100H
171-
:no-value:
172-
173-
WAV2VEC2_ASR_BASE_960H
174-
^^^^^^^^^^^^^^^^^^^^^^
175-
176-
.. container:: py attribute
177-
178-
.. autodata:: WAV2VEC2_ASR_BASE_960H
179-
:no-value:
180-
181-
WAV2VEC2_LARGE
182-
^^^^^^^^^^^^^^
183-
184-
.. container:: py attribute
185-
186-
.. autodata:: WAV2VEC2_LARGE
187-
:no-value:
188-
189-
WAV2VEC2_ASR_LARGE_10M
190-
^^^^^^^^^^^^^^^^^^^^^^
191-
192-
.. container:: py attribute
193-
194-
.. autodata:: WAV2VEC2_ASR_LARGE_10M
195-
:no-value:
196-
197-
WAV2VEC2_ASR_LARGE_100H
198-
^^^^^^^^^^^^^^^^^^^^^^^
199-
200-
.. container:: py attribute
201-
202-
.. autodata:: WAV2VEC2_ASR_LARGE_100H
203-
:no-value:
204-
205-
WAV2VEC2_ASR_LARGE_960H
206-
^^^^^^^^^^^^^^^^^^^^^^^
207-
208-
.. container:: py attribute
209-
210-
.. autodata:: WAV2VEC2_ASR_LARGE_960H
211-
:no-value:
212-
213-
WAV2VEC2_LARGE_LV60K
214-
^^^^^^^^^^^^^^^^^^^^
215-
216-
.. container:: py attribute
217-
218-
.. autodata:: WAV2VEC2_LARGE_LV60K
219-
:no-value:
220-
221-
WAV2VEC2_ASR_LARGE_LV60K_10M
222-
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
223-
224-
.. container:: py attribute
225-
226-
.. autodata:: WAV2VEC2_ASR_LARGE_LV60K_10M
227-
:no-value:
228-
229-
WAV2VEC2_ASR_LARGE_LV60K_100H
230-
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
231-
232-
.. container:: py attribute
233-
234-
.. autodata:: WAV2VEC2_ASR_LARGE_LV60K_100H
235-
:no-value:
236-
237-
WAV2VEC2_ASR_LARGE_LV60K_960H
238-
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
239-
240-
.. container:: py attribute
241-
242-
.. autodata:: WAV2VEC2_ASR_LARGE_LV60K_960H
243-
:no-value:
244-
245-
WAV2VEC2_XLSR53
246-
^^^^^^^^^^^^^^^
247-
248-
.. container:: py attribute
249-
250-
.. autodata:: WAV2VEC2_XLSR53
251-
:no-value:
252-
253-
HUBERT_BASE
254-
^^^^^^^^^^^
255-
256-
.. container:: py attribute
257-
258-
.. autodata:: HUBERT_BASE
259-
:no-value:
260-
261-
HUBERT_LARGE
262-
^^^^^^^^^^^^
263-
264-
.. container:: py attribute
265-
266-
.. autodata:: HUBERT_LARGE
267-
:no-value:
268-
269-
HUBERT_XLARGE
270-
^^^^^^^^^^^^^
271-
272-
.. container:: py attribute
273-
274-
.. autodata:: HUBERT_XLARGE
275-
:no-value:
276-
277-
HUBERT_ASR_LARGE
278-
^^^^^^^^^^^^^^^^
279-
280-
.. container:: py attribute
281-
282-
.. autodata:: HUBERT_ASR_LARGE
283-
:no-value:
284-
285-
HUBERT_ASR_XLARGE
286-
^^^^^^^^^^^^^^^^^
287-
288-
.. container:: py attribute
289-
290-
.. autodata:: HUBERT_ASR_XLARGE
291-
:no-value:
292-
29399
Utility Functions
294100
-----------------
295101

@@ -310,26 +116,12 @@ import_fairseq_model
310116
WaveRNN
311117
~~~~~~~
312118

313-
Model
314-
-----
315-
316-
WaveRNN
317-
^^^^^^^
318-
319119
.. autoclass:: WaveRNN
320120

321121
.. automethod:: forward
322122

323123
.. automethod:: infer
324124

325-
Factory Functions
326-
-----------------
327-
328-
wavernn
329-
^^^^^^^
330-
331-
.. autofunction:: wavernn
332-
333125
References
334126
~~~~~~~~~~
335127

0 commit comments

Comments
 (0)