Teo Wen Shen da87e7fc99

add weights_only=False to torch.load (#1984)

2025-07-10 15:27:08 +08:00

..

__init__.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

asr_datamodule.py

Decrease num_buckets & remove shuffle_buffer_size (#1955)

2025-06-19 12:26:37 +08:00

beam_search.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

conformer.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

decode_aishell.py

Add finetuning script for aishell (#974)

2023-03-30 17:08:46 +08:00

decode.py

add weights_only=False to torch.load (#1984)

2025-07-10 15:27:08 +08:00

decoder.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

encoder_interface.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

export-onnx.py

fixed a CI test for wenetspeech (#1476)

2024-01-27 06:41:56 +08:00

export.py

Fix torchscript export to use tokens.txt instead of lang_dir (#1475)

2024-01-26 19:18:33 +08:00

finetune.py

add weights_only=False to torch.load (#1984)

2025-07-10 15:27:08 +08:00

jit_pretrained.py

Use high_freq -400 in computing fbank features. (#1447)

2024-01-04 13:59:32 +08:00

joiner.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

lstmp.py

Support exporting LSTM with projection to ONNX (#621)

2022-10-18 11:25:31 +08:00

model.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

onnx_check.py

Fix CI tests (#1266)

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

onnx_pretrained.py

add onnx export for stateless2 (#1086)

2023-05-23 16:11:00 +08:00

optim.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

pretrained.py

add weights_only=False to torch.load (#1984)

2025-07-10 15:27:08 +08:00

scaling_converter.py

Support exporting to ONNX for the wenetspeech recipe (#615)

2022-10-13 15:17:20 +08:00

scaling.py

[Ready to merge] Pruned Transducer Stateless2 for WenetSpeech (char-based) (#349)

2022-05-23 17:13:01 +08:00

train.py

Revert "Replace deprecated pytorch methods (#1814)" (#1841)

2024-12-18 16:49:57 +08:00