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

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

asr_datamodule.py

Add Context biasing (#1038)

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

beam_search.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

conformer.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

decode_aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

decode.py

Fix wenetspeech decoding speed (#953)

2023-03-21 21:35:32 +08:00

decoder.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

encoder_interface.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

export-onnx.py

add onnx export for stateless2 (#1086)

2023-05-23 16:11:00 +08:00

export.py

add onnx export for stateless2 (#1086)

2023-05-23 16:11:00 +08:00

finetune.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

jit_pretrained.py

manual correction of black formatting

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

joiner.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

lstmp.py

Support exporting LSTM with projection to ONNX (#621)

2022-10-18 11:25:31 +08:00

model.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

onnx_check.py

apply black on all files

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

onnx_pretrained.py

add onnx export for stateless2 (#1086)

2023-05-23 16:11:00 +08:00

optim.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

pretrained.py

manual correction of black formatting

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

scaling_converter.py

Support exporting to ONNX for the wenetspeech recipe (#615)

2022-10-13 15:17:20 +08:00

scaling.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

train.py

Filter the training data of T < S for Wenet train recipe (#753)

2022-12-11 20:16:10 +08:00