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

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

alignment.py

Support batch-wise forced-alignment (#970)

2023-03-28 23:24:24 +08:00

asr_datamodule.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

beam_search.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

compute_ali.py

doc str fixes (#1241)

2023-09-07 16:34:53 +08:00

decode_gigaspeech.py

Add adaption recipe for pruned_transducer_stateless7 (#1059)

2023-05-17 16:02:27 +08:00

decode.py

Fix typo (#1114)

2023-06-02 14:12:42 +08:00

decoder.py

Update decoder.py (#1262)

2023-09-20 08:15:54 +08:00

encoder_interface.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +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

finetune.py

Fix typo (#1114)

2023-06-02 14:12:42 +08:00

generate_model_from_checkpoint.py

Update generate_model_from_checkpoint.py (#901)

2023-02-13 14:05:38 +08:00

gigaspeech_scoring.py

Support fine-tuning (#944)

2023-03-17 13:44:29 +08:00

gigaspeech.py

Fixes to incorporate with the latest Lhotse release (#1249)

2023-09-13 12:39:49 +08:00

jit_pretrained.py

manual correction of black formatting

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

joiner.py

Export streaming zipformer to ncnn (#906)

2023-02-13 23:41:43 +08:00

model.py

apply black on all files

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

onnx_check.py

Add ONNX support for Zipformer and ConvEmformer (#884)

2023-02-09 00:02:38 +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

Add ONNX support for Zipformer and ConvEmformer (#884)

2023-02-09 00:02:38 +08:00

optim.py

minor fix, remove numel = p.numel() in optim.py (#967)

2023-03-24 15:30:29 +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

scaling_converter.py

Export streaming zipformer to ncnn (#906)

2023-02-13 23:41:43 +08:00

scaling.py

Streaming Zipformer with multi-dataset (#984)

2023-04-21 15:43:28 +08:00

test_compute_ali.py

Support batch-wise forced-alignment (#970)

2023-03-28 23:24:24 +08:00

test_model.py

minor fix for zipformer recipe (#758)

2022-12-13 15:47:30 +08:00

test_onnx.py

Fix CI tests (#1266)

2023-09-21 21:16:14 +08:00

train.py

Use symlinks for best epochs (#1123)

2023-06-12 13:51:46 +08:00

zipformer.py

remove if-branch at downsample pad in zipformer for onnx-export compatibility (#965)

2023-03-24 14:30:43 +08:00