Wei Kang ba257efbcd

Add Context biasing (#1038) ...

* Add context biasing for librispeech

* Add context biasing for wenetspeech

* fix bugs

* Implement Aho-Corasick context graph

* fix some bugs

* Fixes to forward_one_step; add draw to context graph

* add output arc; fix black

* Fix wenetspeech tokenizer

* Minor fixes to the decode.py

2023-06-03 21:28:49 +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 Context biasing (#1038)

2023-06-03 21:28:49 +08:00

conformer.py

add option of using full attention for streaming model decoding (#975)

2023-03-30 14:30:13 +08:00

decode_stream.py

Using streaming conformer as transducer encoder (#380)

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

decode.py

add option of using full attention for streaming model decoding (#975)

2023-03-30 14:30:13 +08:00

decoder.py

Fix padding_idx (#942)

2023-03-10 14:37:28 +08:00

encoder_interface.py

Add remaining files in pruned_transducer_stateless2

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

export.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

joiner.py

Simplify ONNX export (#881)

2023-02-07 15:01:59 +08:00

model.py

apply black on all files

2022-11-17 09:42:17 -05:00

optim.py

apply black on all files

2022-11-17 09:42:17 -05:00

pretrained.py

manual correction of black formatting

2022-11-17 14:18:05 -05:00

scaling.py

Tedlium3 conformer ctc2 (#696)

2022-12-13 16:13:26 +08:00

streaming_beam_search.py

apply black on all files

2022-11-17 09:42:17 -05:00

streaming_decode.py

Fix style issues. (#937)

2023-03-08 22:56:04 +08:00

test_model.py

Using streaming conformer as transducer encoder (#380)

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

train.py

shuffle full Librispeech data (#574)

2022-11-27 11:26:09 +08:00