From b7f4c52e460b9bf3bd85736b2f0de5e7357f19c6 Mon Sep 17 00:00:00 2001
From: Adel Moumen <88119391+Adel-Moumen@users.noreply.github.com>
Date: Sun, 4 Feb 2024 23:36:57 +0100
Subject: [PATCH] check if eos is override by blank index (#2348)
---
speechbrain/decoders/ctc.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/speechbrain/decoders/ctc.py b/speechbrain/decoders/ctc.py
index d25dc1f57..223ddfeb5 100644
--- a/speechbrain/decoders/ctc.py
+++ b/speechbrain/decoders/ctc.py
@@ -232,8 +232,9 @@ class CTCPrefixScore:
self.last_frame_index[i // beam_size], i
]
- # Exclude blank probs for joint scoring
- psi[:, self.blank_index] = self.minus_inf
+ if self.eos_index != self.blank_index:
+ # Exclude blank probs for joint scoring
+ psi[:, self.blank_index] = self.minus_inf
return psi - psi_prev, (r, psi, scoring_table)
--
GitLab