From 2b0573fefb42d57da6c04bfcf71e94ec301b2733 Mon Sep 17 00:00:00 2001 From: Yonnie1331 Date: Sat, 1 Jun 2024 11:16:20 +0900 Subject: [PATCH] unuse forward when jit exporting transducer --- wenet/transducer/transducer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/wenet/transducer/transducer.py b/wenet/transducer/transducer.py index 224077337..ff730cd50 100644 --- a/wenet/transducer/transducer.py +++ b/wenet/transducer/transducer.py @@ -90,6 +90,7 @@ def __init__( normalize_length=length_normalized_loss, ) + @torch.jit.unused def forward( self, batch: dict,