marcoyang1998 52c24df61d

Fix model avg (#1317) ...

* fix a bug about the model_avg during finetuning by exchanging the order of loading pre-trained model and initializing avg model

* only match the exact module prefix

2023-10-18 17:36:14 +08:00

..

__init__.py

Add Zipformer from Dan (#672)

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

alignment.py

formatted the entire LibriSpeech recipe (#1270)

2023-09-24 17:31:01 +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 model avg (#1317)

2023-10-18 17:36:14 +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

Compatibility with the latest Lhotse (#1314)

2023-10-17 21:22:32 +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

fixes for init value of diagnostics.TensorDiagnosticOptions (#1269)

2023-09-24 17:06:47 +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

fixes for init value of diagnostics.TensorDiagnosticOptions (#1269)

2023-09-24 17:06:47 +08:00

zipformer.py

Fix padding issues (#1303)

2023-10-11 16:58:00 +08:00