mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity
icefall/egs/aishell/ASR/pruned_transducer_stateless7_streaming
History
Fangjun Kuang fba5e67d5e
Fix CI tests. (#1974) 
- Introduce unified AMP helpers (create_grad_scaler, torch_autocast) to handle 
  deprecations in PyTorch ≥2.3.0

- Replace direct uses of torch.cuda.amp.GradScaler and torch.cuda.amp.autocast 
  with the new utilities across all training and inference scripts

- Update all torch.load calls to include weights_only=False for compatibility with 
  newer PyTorch versions
2025-07-01 13:47:55 +08:00
..
asr_datamodule.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
beam_search.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
decode_stream.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
decode.py
Enabling char_level and compute_CER for aishell recipe (#1554)
2024-03-18 11:57:47 +08:00
decoder.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
do_not_use_it_directly.py
Revert "Replace deprecated pytorch methods (#1814)" (#1841)
2024-12-18 16:49:57 +08:00
encoder_interface.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
export-for-ncnn-zh.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
export-for-ncnn.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
export-onnx-zh.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
export-onnx.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
export.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
jit_pretrained.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
jit_trace_export.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
jit_trace_pretrained.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
joiner.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
model.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
ncnn_custom_layer.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
onnx_check.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
onnx_model_wrapper.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
onnx_pretrained.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
optim.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
pretrained.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
README.md
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
scaling_converter.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
scaling.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
streaming_beam_search.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
streaming_decode.py
streaming_decode.py, relax the audio range from [-1,+1] to [-10,+10] (#1448)
2024-01-05 10:21:27 +08:00
streaming-ncnn-decode.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
test_model.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00
train.py
Fix CI tests. (#1974)
2025-07-01 13:47:55 +08:00
zipformer_for_ncnn_export_only.py
Rename zipformer2 to zipformer_for_ncnn_export_only to avoid confusion. (#1407)
2023-12-08 14:29:24 +08:00
zipformer.py
[WIP] AISHELL-1 pruned transducer stateless7 streaming recipe (#1300)
2023-10-16 16:28:16 +08:00

README.md

This recipe implements Streaming Zipformer-Transducer model.

See https://k2-fsa.github.io/icefall/recipes/Streaming-ASR/librispeech/zipformer_transducer.html for detailed tutorials.

./emformer.py and ./train.py are basically the same as ./emformer2.py and ./do_not_use_it_directly.py. The only purpose of ./emformer2.py and ./do_not_use_it_directly.py is for exporting to sherpa-ncnn.