Add SD and SV heads for WavLM (#14847)

* Add converted heads

* Add dummies
This commit is contained in:
Anton Lozhkov
2021-12-20 16:40:56 +03:00
committed by GitHub
parent cd583bdaa5
commit 3883e3a75e
10 changed files with 573 additions and 7 deletions

View File

@@ -81,3 +81,17 @@ WavLMForSequenceClassification
.. autoclass:: transformers.WavLMForSequenceClassification
:members: forward
WavLMForAudioFrameClassification
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.WavLMForAudioFrameClassification
:members: forward
WavLMForXVector
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.WavLMForXVector
:members: forward