from local

This commit is contained in:
dohe0342 2023-01-12 16:03:50 +09:00
parent 71cbd39280
commit 400058c6f9
3 changed files with 1 additions and 0 deletions

View File

@ -252,6 +252,7 @@ class Conformer(EncoderInterface):
for enum, alpha in enumerate(self.alpha):
x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1]
x = self.layer_norm(x/self.group_num)
'''
layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs]