Update decoder.py (#1262)

k2-fsa · Sep 20, 2023 · bbb03f7 · bbb03f7
1 parent 7e1288a
commit bbb03f7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py b/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py
@@ -71,6 +71,10 @@ def __init__(
                 groups=decoder_dim // 4,  # group size == 4
                 bias=False,
             )
+        else:
+            # To avoid `RuntimeError: Module 'Decoder' has no attribute 'conv'`
+            # when inference with torch.jit.script and context_size == 1
+            self.conv = nn.Identity()
 
     def forward(self, y: torch.Tensor, need_pad: bool = True) -> torch.Tensor:
         """