Zengwei Yao 6826b076d4

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

* add flops profiler, support for Zipformer encoder and Conformer encoder

* support for reworked conformer and old zipformer

* skip black check

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

Shallow fusion for Aishell (#954)

2023-04-03 16:20:29 +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

Export int8 quantized models for non-streaming Zipformer. (#977)

2023-03-31 22:46:19 +08:00

export.py

Add ONNX support for Zipformer and ConvEmformer (#884)

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

finetune.py

Fix stateless7 training error (#1082)

2023-05-23 12:52:02 +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

multidataset.py

Use CutSet.mux for multidataset (#1020)

2023-04-23 18:41:44 +08: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

manual correction of black formatting

2022-11-17 14:18:05 -05: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

Fix stateless7 training error (#1082)

2023-05-23 12:52:02 +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