Fangjun Kuang 34e40a86b3

Fix exporting decoder model to onnx (#1264) ...

* Use torch.jit.script() to export the decoder model

See also https://github.com/k2-fsa/sherpa-onnx/issues/327

2023-09-22 09:57:15 +08:00

..

__init__.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

asr_datamodule.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

beam_search.py

apply black on all files

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

conformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

decode_stream.py

apply black on all files

2022-11-17 09:42:17 -05: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

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

export-onnx.py

Fix exporting decoder model to onnx (#1264)

2023-09-22 09:57:15 +08:00

export.py

Use tokens.txt to replace bpe.model (#1162)

2023-08-12 16:53:59 +08:00

joiner.py

Add streaming Emformer stateless RNN-T. (#390)

2022-06-01 14:31:47 +08:00

model.py

apply black on all files

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

onnx_check.py

support decoding and computing RTF on test sets with onnx models (#995)

2023-04-12 19:04:50 +08:00

onnx_decode.py

support decoding and computing RTF on test sets with onnx models (#995)

2023-04-12 19:04:50 +08:00

onnx_pretrained.py

support decoding and computing RTF on test sets with onnx models (#995)

2023-04-12 19:04:50 +08:00

pretrained.py

Use tokens.txt to replace bpe.model (#1162)

2023-08-12 16:53:59 +08:00

profile.py

add flops profiler, support for Zipformer encoder and Conformer encoder (#1093)

2023-05-24 19:10:45 +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

subsampling.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00

test_decoder.py

Various fixes to support torch script. (#371)

2022-05-16 21:46:59 +08:00

test_model.py

Apply delay penalty on transducer (#654)

2022-11-04 16:10:09 +08:00

train.py

shuffle full Librispeech data (#574)

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

transformer.py

Use k2 pruned transducer loss to train conformer-transducer model (#194)

2022-02-17 13:33:54 +08:00