Skip to content
Snippets Groups Projects
Unverified Commit b7f4c52e authored by Adel Moumen's avatar Adel Moumen Committed by GitHub
Browse files

check if eos is override by blank index (#2348)

parent ff141ef6
No related branches found
No related tags found
No related merge requests found
...@@ -232,6 +232,7 @@ class CTCPrefixScore: ...@@ -232,6 +232,7 @@ class CTCPrefixScore:
self.last_frame_index[i // beam_size], i self.last_frame_index[i // beam_size], i
] ]
if self.eos_index != self.blank_index:
# Exclude blank probs for joint scoring # Exclude blank probs for joint scoring
psi[:, self.blank_index] = self.minus_inf psi[:, self.blank_index] = self.minus_inf
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment