zr_jin
5c0dfa52d2
Update egs/librispeech/ASR/pruned_transducer_stateless7/export-onnx.py
...
Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>
2023-07-24 23:54:35 +08:00
jinzr
c37ca4dd66
added tests for zipformer
streaming & non-streaming export
...
details are listed below:
1. updated `git lfs` command in `export-onnx.py` and `onnx_pretrained-streaming.py`;
2. added rounding code to `export.py` for case where `params.avg == 1`.
2023-07-24 23:54:35 +08:00
jinzr
83e26a63e3
applied isort
and removed unused imports
2023-07-24 23:54:35 +08:00
jinzr
ccb6031853
moved num_tokens
to utils.py
...
moved `num_tokens` to `icefall/utils.py` to reduce code redundancy
2023-07-24 23:54:35 +08:00
jinzr
f1fe409dee
Update export-onnx.py
...
updated `export-onnx.py` to apply the same change as in [#1152 ](https://github.com/k2-fsa/icefall/pull/1152 )
2023-07-24 23:54:35 +08:00
jinzr
aa2fc799c6
Update export-onnx.py
...
updated `export-oonx.py` to accept `tokens.txt` for blank_id and vocab_size
2023-07-24 23:54:35 +08:00
Yifan Yang
ca60ced213
Fix typo ( #1114 )
...
* Fix typo for zipformer
* Fix typo for pruned_transducer_stateless7
* Fix typo for pruned_transducer_stateless7_ctc
* Fix typo for pruned_transducer_stateless7_ctc_bs
* Fix typo for pruned_transducer_stateless7_streaming
* Fix typo for pruned_transducer_stateless7_streaming_multi
* Fix file permissions for pruned_transducer_stateless7_streaming_multi
* Fix typo for pruned_transducer_stateless8
* Fix typo for pruned_transducer_stateless6
* Fix typo for pruned_transducer_stateless5
* Fix typo for pruned_transducer_stateless4
* Fix typo for pruned_transducer_stateless3
2023-06-02 14:12:42 +08:00
Fangjun Kuang
a632b24c35
Export int8 quantized models for non-streaming Zipformer. ( #977 )
...
* Export int8 quantized models for non-streaming Zipformer.
* Delete export-onnx.py
* Export int8 models for other folders
2023-03-31 22:46:19 +08:00
Fangjun Kuang
8c3ea93fc8
Save meta data to exported ONNX models ( #968 )
2023-03-27 11:39:29 +08:00
Fangjun Kuang
2b995639b7
Add ONNX support for Zipformer and ConvEmformer ( #884 )
2023-02-09 00:02:38 +08:00