jinzr 73e8f62677 added tests for zipformer streaming & non-streaming export ...

details are listed below:
1. updated `git lfs` command in `export-onnx.py` and `onnx_pretrained-streaming.py`;
2. added rounding code to `export.py` for case where `params.avg == 1`.

2023-07-06 22:47:56 +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

Support batch-wise forced-alignment (#970)

2023-03-28 23:24:24 +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

Fix padding_idx (#942)

2023-03-10 14:37:28 +08:00

encoder_interface.py

Add Zipformer from Dan (#672)

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

export-onnx.py

added tests for zipformer streaming & non-streaming export

2023-07-06 22:47:56 +08:00

export.py

applied isort and removed unused imports

2023-07-06 15:16:32 +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

Support fine-tuning (#944)

2023-03-17 13:44:29 +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

fixed file permission

2023-07-04 15:02:39 +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

Add Zipformer Onnx Support (#778)

2023-01-03 16:59:44 +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