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

..

aishell.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

asr_datamodule.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

beam_search.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

decode.py

Add Context biasing (#1038)

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

decoder2.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

decoder.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

encoder_interface.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

export-onnx.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

export.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

jit_pretrained.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

joiner.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

model.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

onnx_check.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

onnx_pretrained.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

optim.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

pretrained.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

scaling_converter.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

scaling.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00

train2.py

Remove cur_batch_idx (#1102)

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

train.py

Remove cur_batch_idx (#1102)

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

zipformer.py

Aishell pruned_transducer_stateless7 (#962)

2023-05-23 11:04:33 +08:00