mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 17:42:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

add @torch.jit.export for streaming_forward func in Zipformer class (#1004)

Browse Source
This commit is contained in:
Zengwei Yao 2023-04-17 12:03:52 +08:00 committed by GitHub
parent 5f066d3d53
commit 7c7d9ab042
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/pruned_transducer_stateless7_streaming/zipformer.py
View File

@ -570,6 +570,7 @@ class Zipformer(EncoderInterface):
return x, lengths return x, lengths
@torch.jit.export
def streaming_forward( def streaming_forward(
self, self,
x: torch.Tensor, x: torch.Tensor,