models.rst 1.84 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
Model
-----

hwangjeff's avatar
hwangjeff committed
34
35
Tacotron2
^^^^^^^^^
moto's avatar
moto committed
36

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
-----------------

78
79
80
81
82
83
wav2vec2_model
^^^^^^^^^^^^^^

.. autofunction:: wav2vec2_model


moto's avatar
moto committed
84
wav2vec2_base
moto's avatar
moto committed
85
^^^^^^^^^^^^^
moto's avatar
moto committed
86

moto's avatar
moto committed
87
88
.. autofunction:: wav2vec2_base

moto's avatar
moto committed
89
wav2vec2_large
moto's avatar
moto committed
90
^^^^^^^^^^^^^^
moto's avatar
moto committed
91

moto's avatar
moto committed
92
93
.. autofunction:: wav2vec2_large

moto's avatar
moto committed
94
wav2vec2_large_lv60k
moto's avatar
moto committed
95
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
96

moto's avatar
moto committed
97
98
.. autofunction:: wav2vec2_large_lv60k

moto's avatar
moto committed
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
hubert_base
^^^^^^^^^^^

.. autofunction:: hubert_base

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

.. autofunction:: hubert_large

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

.. autofunction:: hubert_xlarge

114
115
116
Utility Functions
-----------------

117
118
.. currentmodule:: torchaudio.models.wav2vec2.utils

moto's avatar
moto committed
119
import_huggingface_model
moto's avatar
moto committed
120
^^^^^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
121

122
123
.. autofunction:: import_huggingface_model

moto's avatar
moto committed
124
import_fairseq_model
moto's avatar
moto committed
125
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
126

moto's avatar
moto committed
127
128
.. autofunction:: import_fairseq_model

129
.. currentmodule:: torchaudio.models
moto's avatar
moto committed
130

moto's avatar
moto committed
131
132
WaveRNN
~~~~~~~
jimchen90's avatar
jimchen90 committed
133

moto's avatar
moto committed
134
135
136
137
138
139
Model
-----

WaveRNN
^^^^^^^

jimchen90's avatar
jimchen90 committed
140
141
142
.. autoclass:: WaveRNN

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

144
145
  .. automethod:: infer

146
147
148
149
Factory Functions
-----------------

wavernn
moto's avatar
moto committed
150
^^^^^^^
151
152
153

.. autofunction:: wavernn

moto's avatar
moto committed
154
155
156
157
References
~~~~~~~~~~

.. footbibliography::