ezerhouni 9b671e1c21

Add Shallow fusion in modified_beam_search (#630) ...

* Add utility for shallow fusion

* test batch size == 1 without shallow fusion

* Use shallow fusion for modified-beam-search

* Modified beam search with ngram rescoring

* Fix code according to review

Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>

2022-10-21 16:44:56 +08:00

..

__init__.py

Start adding some files..

2022-03-16 22:14:30 +08:00

asr_datamodule.py

Start adding some files..

2022-03-16 22:14:30 +08:00

beam_search.py

Add Shallow fusion in modified_beam_search (#630)

2022-10-21 16:44:56 +08:00

conformer.py

Modify ActivationBalancer for speed (#612)

2022-10-13 15:14:28 +08:00

decode_stream.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

decode.py

fix type hints for decode.py (#623)

2022-10-18 06:56:12 +08:00

decoder.py

Sort results to make it more convenient to compare decoding results (#522)

2022-08-12 07:12:50 +08:00

encoder_interface.py

Add remaining files in pruned_transducer_stateless2

2022-03-17 11:17:52 +08:00

export.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

joiner.py

Sort results to make it more convenient to compare decoding results (#522)

2022-08-12 07:12:50 +08:00

model.py

Use ScaledLSTM as streaming encoder (#479)

2022-08-19 14:38:45 +08:00

optim.py

Add clamping operation in Eve optimizer for all scalar weights to avoid (#550)

2022-08-25 12:12:50 +08:00

pretrained.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

scaling.py

Modify ActivationBalancer for speed (#612)

2022-10-13 15:14:28 +08:00

streaming_beam_search.py

Add modified_beam_search for streaming decode (#489)

2022-07-25 16:53:23 +08:00

streaming_decode.py

Sort results to make it more convenient to compare decoding results (#522)

2022-08-12 07:12:50 +08:00

test_model.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

train.py

Use ScaledLSTM as streaming encoder (#479)

2022-08-19 14:38:45 +08:00