Unverified Commit b57f05c4 authored by Vincent QB's avatar Vincent QB Committed by GitHub
Browse files

add doc for rnnt loss (#1171)

parent 1bb1115f
...@@ -39,6 +39,7 @@ The :mod:`torchaudio` package consists of I/O, popular datasets and common audio ...@@ -39,6 +39,7 @@ The :mod:`torchaudio` package consists of I/O, popular datasets and common audio
compliance.kaldi compliance.kaldi
kaldi_io kaldi_io
utils utils
transducer
.. toctree:: .. toctree::
......
.. role:: hidden
:class: hidden-section
torchaudio.prototype.transducer
===============================
.. currentmodule:: torchaudio.prototype.transducer
.. note::
The RNN transducer loss is a prototype feature, see `here <https://pytorch.org/audio>`_ to learn more about the nomenclature. It is only available within the nightlies, and also needs to be imported explicitly using: :code:`from torchaudio.prototype.transducer import rnnt_loss, RNNTLoss`.
rnnt_loss
---------
.. autofunction:: rnnt_loss
:hidden:`RNNTLoss`
~~~~~~~~~~~~~~~~~~
.. autoclass:: RNNTLoss
.. automethod:: forward
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment