from local

This commit is contained in:
dohe0342 2023-02-15 13:07:09 +09:00
parent df013c37fa
commit f407e7cfea
3 changed files with 1 additions and 1 deletions

View File

@ -189,7 +189,7 @@ class Conformer(Transformer):
ctc_output=self.ctc_output,
) # (S, N, C)
if self.group_num != 0:
if self.group_num > 0:
x = 0
for enum, alpha in enumerate(self.alpha):
x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1]