Import torchaudio #1890 211270d

Reviewed By: mthrok Differential Revision: D31728916 fbshipit-source-id: 4b932f285c274c5f2197325ce73ecbd930e3597e
pytorch · Oct 21, 2021 · b2f91c2 · b2f91c2
1 parent 1910dbc
commit b2f91c2
Show file tree

Hide file tree

Showing 50 changed files with 2,258 additions and 1,797 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in
@@ -517,7 +517,7 @@ jobs:
       name: windows-gpu
     environment:
       <<: *environment
-      CUDA_VERSION: "10.2"
+      CUDA_VERSION: "11.3"
       TORCHAUDIO_TEST_FORCE_CUDA: 1
     steps:
       - checkout

diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py
@@ -22,7 +22,7 @@
 
 PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"]
 CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111","cu113", "rocm4.1"],
-                    "windows": ["cpu", "cu102", "cu111"],
+                    "windows": ["cpu", "cu113"],
                     "macos": ["cpu"]}
 
 
@@ -209,7 +209,7 @@ def unittest_workflows(indentation=6):
                 job = {
                     "name": f"unittest_{os_type}_{device_type}_py{python_version}",
                     "python_version": python_version,
-                    "cuda_version": 'cpu' if device_type=="cpu" else "cu102",
+                    "cuda_version": 'cpu' if device_type == "cpu" else "cu113",
                 }
 
                 if os_type != "windows":
@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6):
                         "stylecheck": {
                             "name": f"stylecheck_py{python_version}",
                             "python_version": python_version,
-                            "cuda_version": 'cpu' if device_type=="cpu" else "cu102",
+                            "cuda_version": "cpu",
                         }
                     })
     return indent(indentation, jobs)

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -35,6 +35,7 @@ The :mod:`torchaudio` package consists of I/O, popular datasets and common audio
    transforms
    datasets
    models
+   pipelines
    sox_effects
    compliance.kaldi
    kaldi_io

diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -28,27 +28,12 @@ DeepSpeech
 Tacotron2
 ~~~~~~~~~
 
-Model
------
-
-Tacotoron2
-^^^^^^^^^^
-
 .. autoclass:: Tacotron2
 
   .. automethod:: forward
 
   .. automethod:: infer
 
-Factory Functions
------------------
-
-tacotron2
-^^^^^^^^^
-
-.. autofunction:: tacotron2
-
-
 Wav2Letter
 ~~~~~~~~~~
 
@@ -96,21 +81,6 @@ wav2vec2_large_lv60k
 
 .. autofunction:: wav2vec2_large_lv60k
 
-wav2vec2_ft_base
-^^^^^^^^^^^^^^^^^
-
-.. autofunction:: wav2vec2_ft_base
-
-wav2vec2_ft_large
-^^^^^^^^^^^^^^^^^^
-
-.. autofunction:: wav2vec2_ft_large
-
-wav2vec2_ft_large_lv60k
-^^^^^^^^^^^^^^^^^^^^^^^^
-
-.. autofunction:: wav2vec2_ft_large_lv60k
-
 hubert_base
 ^^^^^^^^^^^
 
@@ -126,170 +96,6 @@ hubert_xlarge
 
 .. autofunction:: hubert_xlarge
 
-hubert_ft_large
-^^^^^^^^^^^^^^^^
-
-.. autofunction:: hubert_ft_large
-
-hubert_ft_xlarge
-^^^^^^^^^^^^^^^^^
-
-.. autofunction:: hubert_ft_xlarge
-
-Pre-trained Models
-------------------
-
-.. autoclass:: Wav2Vec2PretrainedModelBundle
-
-   .. automethod:: get_model
-
-   .. autoproperty:: labels
-
-
-WAV2VEC2_BASE
-^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_BASE
-      :no-value:
-
-WAV2VEC2_ASR_BASE_10M
-^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: torchaudio.models.WAV2VEC2_ASR_BASE_10M
-      :no-value:
-
-WAV2VEC2_ASR_BASE_100H
-^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_BASE_100H
-      :no-value:
-
-WAV2VEC2_ASR_BASE_960H
-^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_BASE_960H
-      :no-value:
-
-WAV2VEC2_LARGE
-^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_LARGE
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_10M
-^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_10M
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_100H
-^^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_100H
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_960H
-^^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_960H
-      :no-value:
-
-WAV2VEC2_LARGE_LV60K
-^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_LARGE_LV60K
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_LV60K_10M
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_10M
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_LV60K_100H
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_100H
-      :no-value:
-
-WAV2VEC2_ASR_LARGE_LV60K_960H
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_ASR_LARGE_LV60K_960H
-      :no-value:
-
-WAV2VEC2_XLSR53
-^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: WAV2VEC2_XLSR53
-      :no-value:
-
-HUBERT_BASE
-^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: HUBERT_BASE
-      :no-value:
-
-HUBERT_LARGE
-^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: HUBERT_LARGE
-      :no-value:
-
-HUBERT_XLARGE
-^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: HUBERT_XLARGE
-      :no-value:
-
-HUBERT_ASR_LARGE
-^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: HUBERT_ASR_LARGE
-      :no-value:
-
-HUBERT_ASR_XLARGE
-^^^^^^^^^^^^^^^^^
-
-.. container:: py attribute
-
-   .. autodata:: HUBERT_ASR_XLARGE
-      :no-value:
-
 Utility Functions
 -----------------
 
@@ -310,26 +116,12 @@ import_fairseq_model
 WaveRNN
 ~~~~~~~
 
-Model
------
-
-WaveRNN
-^^^^^^^
-
 .. autoclass:: WaveRNN
 
   .. automethod:: forward
 
   .. automethod:: infer
 
-Factory Functions
------------------
-
-wavernn
-^^^^^^^
-
-.. autofunction:: wavernn
-
 References
 ~~~~~~~~~~