danqing fu
d932ed0928
onnx streaming decode
2023-06-11 13:42:16 +08:00
danqing fu
0b63b87092
stream onnx export suc
2023-06-08 13:33:38 +08:00
danqing fu
1378f833bd
small fixes
2023-06-08 10:40:15 +08:00
danqing fu
851bb50cb8
small fixes
2023-06-08 09:47:00 +08:00
danqing fu
4d4188aa81
streaming zipformer onnx export
2023-06-07 21:23:11 +08:00
Fangjun Kuang
26f3049f4c
minor fixes
2023-06-06 15:30:14 +08:00
Fangjun Kuang
97b7e34dc5
small fixes
2023-06-06 15:27:23 +08:00
Fangjun Kuang
d9e7f02225
Use torch.jit.script() for positional encoding
2023-06-06 15:25:11 +08:00
Fangjun Kuang
4cb2395186
Support longer input for the offline model
2023-06-06 14:20:12 +08:00
danqing fu
acbc4b5808
minor fixes
2023-06-06 11:13:04 +08:00
danqing fu
e609dfee29
minor fixes
2023-06-06 11:12:21 +08:00
danqing fu
09c4946c15
modify max_len to allow longer input audio
2023-06-06 11:09:00 +08:00
danqing fu
8f8b09498f
remove if-branch at downsampling pad
2023-06-06 10:05:59 +08:00
Fangjun Kuang
7b00b34617
minor fixes
2023-06-02 14:23:47 +08:00
Fangjun Kuang
dd91c89f28
Fix exporting non-streaming zipformer to ONNX via torch.jit.trace()
2023-06-02 14:21:28 +08:00
danqing fu
f43c44236f
add updated zipformer nonstreaming onnx export
2023-06-01 03:36:11 +00:00
Fangjun Kuang
7b0afbdc16
Remove cur_batch_idx ( #1102 )
2023-05-30 14:49:54 +08:00
Zengwei Yao
6826b076d4
add flops profiler, support for Zipformer encoder and Conformer encoder ( #1093 )
...
* add flops profiler, support for Zipformer encoder and Conformer encoder
* support for reworked conformer and old zipformer
* skip black check
2023-05-24 19:10:45 +08:00
Zengwei Yao
30fcd16c7d
rm zipformer/__init__.py ( #1075 )
2023-05-20 23:12:11 +08:00
Zengwei Yao
f18b539fbc
Add the upgraded Zipformer model ( #1058 )
...
* add the zipformer codes, copied from branch from_dan_scaled_adam_exp1119
* support model export with torch.jit.script
* update RESULTS.md
* support exporting streaming model with torch.jit.script
* add results of streaming models, with some minor changes
* update README.md
* add CI test
* update k2 version in requirements-ci.txt
* update pyproject.toml
2023-05-19 16:47:59 +08:00
