Wei Kang 219bba1310

zipformer wenetspeech (#1130) ...

* copy files

* update train.py

* small fixes

* Add decode.py

* Fix dataloader in decode.py

* add blank penalty

* Add blank-penalty to other decoding method

* Minor fixes

* add zipformer2 recipe

* Minor fixes

* Remove pruned7

* export and test models

* Replace bpe with tokens in export.py and pretrain.py

* Minor fixes

* Minor fixes

* Minor fixes

* Fix export

* Update results

* Fix zipformer-ctc

* Fix ci

* Fix ci

* Fix CI

* Fix CI

---------

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

2023-06-26 09:33:18 +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

zipformer wenetspeech (#1130)

2023-06-26 09:33:18 +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