Fix loss returned by CTC model in RNNT (#2327)

wenet-e2e · Jan 26, 2024 · 0c8945c · 0c8945c
1 parent baaa27a
commit 0c8945c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wenet/transducer/transducer.py b/wenet/transducer/transducer.py
@@ -130,8 +130,8 @@ def forward(
         # optional ctc
         loss_ctc: Optional[torch.Tensor] = None
         if self.ctc_weight != 0.0 and self.ctc is not None:
-            loss_ctc = self.ctc(encoder_out, encoder_out_lens, text,
-                                text_lengths)
+            loss_ctc, _ = self.ctc(encoder_out, encoder_out_lens, text,
+                                   text_lengths)
         else:
             loss_ctc = None