from local

This commit is contained in:
dohe0342 2023-01-08 18:57:27 +09:00
parent 381bdda8fc
commit 9fea943791
3 changed files with 2 additions and 2 deletions

Binary file not shown.

View File

@ -201,11 +201,11 @@ class Conformer(EncoderInterface):
) # (T, N, C)
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
layer_output = [x.permute(1, 0, 2) for x in layer_output]
layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs]
x = 0
for enum, alpha in enumerate(self.alpha):
x += self.sigmoid(alpha*layer_output[(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/self.group_num)