9 Commits

Author SHA1 Message Date
jinzr
73e8f62677 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-06 22:47:56 +08:00
jinzr
3a85f49c1a applied isort and removed unused imports 2023-07-06 15:16:32 +08:00
jinzr
f5257b1528 moved num_tokens to utils.py
moved `num_tokens` to `icefall/utils.py` to reduce code redundancy
2023-07-06 12:41:29 +08:00
jinzr
628c608d5c 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-04 14:01:28 +08:00
jinzr
95ec9efcbd Update export-onnx.py
updated `export-oonx.py` to accept `tokens.txt` for blank_id and vocab_size
2023-07-04 12:03:57 +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