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

moto's avatar
moto committed
61
62
Wav2Vec2.0
~~~~~~~~~~
moto's avatar
moto committed
63

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

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

.. autoclass:: Wav2Vec2Model

  .. automethod:: extract_features

  .. automethod:: forward

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

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

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

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

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

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

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

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

.. autofunction:: wav2vec2_asr_base

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

.. autofunction:: wav2vec2_asr_large

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

.. autofunction:: wav2vec2_asr_large_lv60k

109
110
111
112
113
.. currentmodule:: torchaudio.models.wav2vec2.utils

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

moto's avatar
moto committed
114
import_huggingface_model
moto's avatar
moto committed
115
^^^^^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
116
		   
117
118
.. autofunction:: import_huggingface_model

moto's avatar
moto committed
119
import_fairseq_model
moto's avatar
moto committed
120
^^^^^^^^^^^^^^^^^^^^
moto's avatar
moto committed
121
		   
moto's avatar
moto committed
122
123
.. autofunction:: import_fairseq_model

124
.. currentmodule:: torchaudio.models
moto's avatar
moto committed
125

moto's avatar
moto committed
126
127
WaveRNN
~~~~~~~
jimchen90's avatar
jimchen90 committed
128

moto's avatar
moto committed
129
130
131
132
133
134
Model
-----

WaveRNN
^^^^^^^

jimchen90's avatar
jimchen90 committed
135
136
137
.. autoclass:: WaveRNN

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

139
140
  .. automethod:: infer

141
142
143
144
Factory Functions
-----------------

wavernn
moto's avatar
moto committed
145
^^^^^^^
146
147
148

.. autofunction:: wavernn

moto's avatar
moto committed
149
150
151
152
References
~~~~~~~~~~

.. footbibliography::