763 Commits

Author SHA1 Message Date
Yifan Yang
00ed2b7567
Update decoder.py 2023-02-09 18:21:52 +08:00
Yifan Yang
bcec8465c9
Update model.py 2023-02-09 18:21:21 +08:00
Yifan Yang
7c47efd3e1
Merge branch 'k2-fsa:master' into repeat-k 2023-02-09 15:44:11 +08:00
Yifan Yang
5cd1636cb3
Fix a bug in decode.py (#893)
Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
2023-02-09 12:12:23 +08:00
Fangjun Kuang
e916027bfe
Fix doc typos for onnx export (#891) 2023-02-09 10:33:40 +08:00
Karel Vesely
35e5a2475c
Librispeech, validate_manifest.py (#890) 2023-02-09 07:57:02 +08:00
Fangjun Kuang
2b995639b7
Add ONNX support for Zipformer and ConvEmformer (#884) 2023-02-09 00:02:38 +08:00
Zengwei Yao
af735eb75b
Get alignments using lhotse workflows align-with-torchaudio (#888)
* add lhotse workflow align-with-torchaudio

* modify related decode.py files
2023-02-08 21:54:35 +08:00
Yifan Yang
e3beb93e1d
Merge branch 'k2-fsa:master' into repeat-k 2023-02-08 10:51:51 +08:00
Zengwei Yao
d12e6f098c
Get (start, end) timestamps for CTC models (#876)
* parse timestamps and texts for BPE-based models

* parse timestamps (frame indexes) and texts for other cases

* add test functions

* add parse_fsa_timestamps_and_texts function, test in conformer_ctc3/decode.py

* calculate symbol delay for (start, end) timestamps
2023-02-07 21:43:16 +08:00
Yifan Yang
072457f305
Merge branch 'k2-fsa:master' into repeat-k 2023-02-07 19:44:27 +08:00
yifanyang
1551d7e826 modify the implement of computing k 2023-02-07 19:43:51 +08:00
yifanyang
6e377c6874 modify the implement of computing k 2023-02-07 19:33:59 +08:00
Fangjun Kuang
7ae03f6c88
Add onnx export support for pruned_transducer_stateless5 (#883) 2023-02-07 17:47:08 +08:00
Yifan Yang
ffbf6d9199
Add generate_averaged_model.py (#882) 2023-02-07 16:19:08 +08:00
Fangjun Kuang
8d3810e289
Simplify ONNX export (#881)
* Simplify ONNX export

* Fix ONNX CI tests
2023-02-07 15:01:59 +08:00
Fangjun Kuang
52f3a747be
Refactor onnx export for streaming zipformer (#879) 2023-02-07 12:12:26 +08:00
Zengwei Yao
5a05b95730
add params.hlg_scale (#880) 2023-02-06 23:21:46 +08:00
yifanyang
06b583ee84 Fix for style check 2023-02-06 18:51:43 +08:00
yifanyang
3554ed32db Fix for style check 2023-02-06 18:47:36 +08:00
yifanyang
5cf4e2df07 Fix for style check 2023-02-06 18:42:45 +08:00
yifanyang
f43ea5db30 Fix a bug in beam_search 2023-02-06 17:40:01 +08:00
yifanyang
59daf83b59 Add greedy_search_batch 2023-02-06 16:57:46 +08:00
Yifan Yang
51812b2ddc
Merge branch 'k2-fsa:master' into repeat-k 2023-02-06 16:55:06 +08:00
Yifan Yang
caf23546ed
No more T < S after frame_reducer (#875)
* No more T < S after frame_reducer

* Fix for style check

* Adjust the permissions

* Add support for inference to frame_reducer

* Fix for flake8 check

---------

Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
2023-02-06 12:17:45 +08:00
Yuekai Zhang
bf5f0342a2
Add streaming onnx export for zipformer (#831)
* add streaming onnx export for zipformer

* update triton support

* add comments

* add ci test

* add onnxmltools for fp16 onnx export
2023-02-06 10:37:07 +08:00
Yifan Yang
355ecea60b
Merge branch 'k2-fsa:master' into repeat-k 2023-02-05 15:29:30 +08:00
Yifan Yang
029c8566e4
Small fix for frame_reducer.py (#871) 2023-02-03 17:49:54 +08:00
Yifan Yang
bffce413f0
Fix filename ctc_guild_decode_bs.py -> ctc_guide_decode_bs.py (#870)
* fix filename ctc_guild_decode_bs.py -> ctc_guide_decode_bs.py

---------

Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
2023-02-03 12:32:06 +08:00
yifanyang
fbe11bd5a8 fix for style check 2023-02-03 12:24:33 +08:00
Yifan Yang
b27c11c867
Merge branch 'k2-fsa:master' into repeat-k 2023-02-03 11:59:22 +08:00
Zengwei Yao
1e6d6f8160
shuffle full Librispeech for zipformer recipes (#869)
* shuffle libri
2023-02-03 11:54:57 +08:00
yifanyang
76eee0ee80 Fix for style check 2023-02-02 17:51:08 +08:00
yifanyang
c3e01e141f Add repeat_param to embedding_out during training and decoding 2023-02-02 17:33:52 +08:00
yifanyang
45b68e1df6 Add .py 2023-02-02 15:34:20 +08:00
yifanyang
07aed16b96 Add pruned_transducer_stateless9 2023-02-02 15:30:34 +08:00
Yifan Yang
e36ea89112
update result.md for pruned_transducer_stateless7_ctc_bs (#865) 2023-02-01 21:04:56 +08:00
Yifan Yang
d8234e199c
Add export to ONNX for Zipformer+CTC using blank skip (#861)
* Add export to ONNX for Zipformer+CTC using blank skip

---------

Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
2023-01-31 15:57:03 +08:00
BuaaAlban
e9019511eb
Fix bug in streaming_conformer_ctc egs (#862)
* Update train.py

Fix transducer lstm egs bug as mentioned in issue 579

* Update train.py

fix dataloader bug
2023-01-31 15:19:50 +08:00
Yifan Yang
e277e31e37
update huggingface link of zipformer_ctc_blankskip.rst (#858)
* update huggingface link

* update link

---------

Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
2023-01-29 15:35:36 +08:00
Meng Wei
74a2069f94
fix expired links (#856) 2023-01-28 14:43:47 +08:00
Teo Wen Shen
1ce2bc1ee0
edit comments (#852) 2023-01-28 13:47:21 +08:00
Zengwei Yao
6b1ab71dc9
hardcode --filter-uneven-sized-batch (#854) 2023-01-27 21:24:12 +08:00
Wei Kang
f5ff7a18eb
Fix the unclear description for streaming model (#849) 2023-01-17 11:28:59 +08:00
Fangjun Kuang
0af3e7beda
fix export for stateless4 (#844) 2023-01-16 20:26:36 +08:00
Zengwei Yao
2a463a420d
Filter uneven-sized batch (#843)
* add filter_uneven_sized_batch fucntion

* set --filter-uneven-sized-batch=True as default
2023-01-16 20:15:35 +08:00
Fangjun Kuang
5c8e9628cc
update faq for libpython3.10.so not found (#838) 2023-01-13 15:21:29 +08:00
Fangjun Kuang
958dbb3a1d
add doc for int8 quantization with sherpa-ncnn (#832)
* add doc for int8 quantization with sherpa-ncnn

* typo fixes
2023-01-11 20:29:36 +08:00
marcoyang1998
142420b3af
Add docs for distillation (#812)
* add README to docs

* update documents for distillation

* upload png files
2023-01-11 16:45:24 +08:00
Fangjun Kuang
8582b6e41a
Add doc about converting conv-emformer to sherpa-ncnn (#830) 2023-01-11 15:34:30 +08:00