Fangjun Kuang
|
d4d4f281ec
|
Revert "Replace deprecated pytorch methods (#1814)" (#1841)
This reverts commit 3e4da5f78160d3dba3bdf97968bd7ceb8c11631f.
|
2024-12-18 16:49:57 +08:00 |
|
Li Peng
|
3e4da5f781
|
Replace deprecated pytorch methods (#1814)
* Replace deprecated pytorch methods
- torch.cuda.amp.GradScaler(...) => torch.amp.GradScaler("cuda", ...)
- torch.cuda.amp.autocast(...) => torch.amp.autocast("cuda", ...)
* Replace `with autocast(...)` with `with autocast("cuda", ...)`
Co-authored-by: Li Peng <lipeng@unisound.ai>
|
2024-12-16 10:24:16 +08:00 |
|
Wen Ding
|
2b3c5d799f
|
Fix padding issues (#1303)
|
2023-10-11 16:58:00 +08:00 |
|
Peng He
|
f260a09ed4
|
remove if-branch at downsample pad in zipformer for onnx-export compatibility (#965)
|
2023-03-24 14:30:43 +08:00 |
|
Fangjun Kuang
|
2b995639b7
|
Add ONNX support for Zipformer and ConvEmformer (#884)
|
2023-02-09 00:02:38 +08:00 |
|
Teo Wen Shen
|
1ce2bc1ee0
|
edit comments (#852)
|
2023-01-28 13:47:21 +08:00 |
|
Yunusemre
|
0f26edfde9
|
Add Zipformer Onnx Support (#778)
* add export script
* add zipformer onnx pretrained script
* add onnx zipformer test
* fix style
* add zipformer onnx to workflow
* replace is_in_onnx_export with is_tracing
* add github.event.label.name == 'onnx'
* add is_tracing to necessary conditions
* fix pooling_mask
* add onnx_check
* add onnx_check to scripts
* add is_tracing to scaling.py
|
2023-01-03 16:59:44 +08:00 |
|
Zengwei Yao
|
d167aad4ab
|
Add streaming zipformer (#787)
* add streaming zipformer codes
* add test_model.py
* add export.py, pretrained.py, jit_pretrained.py
* add cached_len for pooling module
* add jit_trace_export.py and jit_trace_pretrained.py
* fix bug in jit.trace
* update RESULTS.md
* add CI test
* minor fix in pruned_transducer_stateless7/zipformer.py
* update README.md
|
2022-12-30 10:52:18 +08:00 |
|
Zengwei Yao
|
65d7192dca
|
Fix zipformer attn_output_weights (#774)
* fix attn_output_weights
* remove in-place op
|
2022-12-19 20:10:39 +08:00 |
|
Zengwei Yao
|
fbc1d3b194
|
fix src_key_padding_mask in DownsampledZipformerEncoder (#768)
|
2022-12-17 22:03:13 +08:00 |
|
Zengwei Yao
|
0470bbae66
|
minor fix for zipformer recipe (#758)
* minor fix
* add CI test
|
2022-12-13 15:47:30 +08:00 |
|
Yifan Yang
|
02c18ba4b2
|
rm the dup line of Zipformer.py (#755)
Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
|
2022-12-10 19:34:19 +08:00 |
|
Desh Raj
|
107df3b115
|
apply black on all files
|
2022-11-17 09:42:17 -05:00 |
|
Fangjun Kuang
|
60317120ca
|
Revert "Apply new Black style changes"
|
2022-11-17 20:19:32 +08:00 |
|
Desh Raj
|
d110b04ad3
|
apply new black formatting to all files
|
2022-11-16 13:06:43 -05:00 |
|
Desh Raj
|
c8ce243255
|
Zipformer output length (#686)
* add assertion for output length
* add comment in filter_cuts
* add length filter to Zipformer recipes
|
2022-11-16 11:29:45 +08:00 |
|
Fangjun Kuang
|
7e82f87126
|
Add Zipformer from Dan (#672)
|
2022-11-12 18:11:19 +08:00 |
|