Fangjun Kuang 896993714b Add tests to ensure that the model is torch scriptable.

2022-05-16 19:12:34 +08:00

..

__init__.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

asr_datamodule.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

beam_search.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

conformer.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

decode.py

Ignore padding frames during RNN-T decoding. (#358)

2022-05-13 07:39:14 +08:00

decoder.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

encoder_interface.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

export.py

Various fixes to support torch script.

2022-05-16 18:53:56 +08:00

joiner.py

Add tests to ensure that the model is torch scriptable.

2022-05-16 19:12:34 +08:00

model.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

pretrained.py

Ignore padding frames during RNN-T decoding. (#358)

2022-05-13 07:39:14 +08:00

subsampling.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

test_model.py

Add tests to ensure that the model is torch scriptable.

2022-05-16 19:12:34 +08:00

train.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00

transformer.py

Support computing RNN-T loss with torchaudio (#316)

2022-04-19 18:47:13 +08:00