from local

This commit is contained in:
dohe0342 2023-02-14 12:57:52 +09:00
parent 4efa6c14b0
commit 28c3ef7c24
2 changed files with 5 additions and 2 deletions

View File

@ -183,8 +183,11 @@ class Conformer(Transformer):
for enum, alpha in enumerate(self.alpha): for enum, alpha in enumerate(self.alpha):
x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1] x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1]
x = self.layer_norm(x) x = self.layer_norm(x)
return x, mask if self.interctc:
return layer_outputs, mask
else:
return x, mask
class ConformerEncoderLayer(nn.Module): class ConformerEncoderLayer(nn.Module):