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

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

asr_datamodule.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

beam_search.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

conformer.py

fix bug (#796)

2022-12-28 11:20:38 +08:00

decode_stream.py

apply black on all files

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

decode.py

Add Context biasing (#1038)

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

decoder.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

encoder_interface.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

export-onnx-streaming.py

Add onnx export scripts for wenetspeech recipe. (#1085)

2023-05-23 13:32:14 +08:00

export-onnx.py

Add onnx export scripts for wenetspeech recipe. (#1085)

2023-05-23 13:32:14 +08:00

export.py

fix wenet stateless5 jit export error (#735)

2022-12-05 23:35:10 +08:00

joiner.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

lstmp.py

fix wenet stateless5 jit export error (#735)

2022-12-05 23:35:10 +08:00

model.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

onnx_check.py

Add onnx export scripts for wenetspeech recipe. (#1085)

2023-05-23 13:32:14 +08:00

onnx_pretrained-streaming.py

Add onnx export scripts for wenetspeech recipe. (#1085)

2023-05-23 13:32:14 +08:00

onnx_pretrained.py

Add onnx export scripts for wenetspeech recipe. (#1085)

2023-05-23 13:32:14 +08:00

optim.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +08:00

pretrained.py

manual correction of black formatting

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

scaling_converter.py

fix wenet stateless5 jit export error (#735)

2022-12-05 23:35:10 +08:00

scaling.py

[WIP] Pruned-transducer-stateless5-for-WenetSpeech (offline and streaming) (#447)

2022-07-28 12:54:27 +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

train.py

Remove cur_batch_idx (#1102)

2023-05-30 14:49:54 +08:00