Fangjun Kuang f6ce135608

Various fixes to support torch script. (#371) ...

* Various fixes to support torch script.

* Add tests to ensure that the model is torch scriptable.

* Update tests.

2022-05-16 21:46:59 +08:00

..

__init__.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

asr_datamodule.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

beam_search.py

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

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

conformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

decode.py

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

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

decoder.py

Tedlium3 pruned transducer stateless (#261)

2022-04-11 17:08:53 +08:00

encoder_interface.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

export.py

Various fixes to support torch script. (#371)

2022-05-16 21:46:59 +08:00

joiner.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

model.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

pretrained.py

Update results for libri+giga multi dataset setup. (#363)

2022-05-14 21:45:39 +08:00

subsampling.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

test_decoder.py

Various fixes to support torch script. (#371)

2022-05-16 21:46:59 +08:00

test_model.py

Various fixes to support torch script. (#371)

2022-05-16 21:46:59 +08:00

train.py

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

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

transformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00