from local

This commit is contained in:
dohe0342 2023-02-01 17:43:18 +09:00
parent 600ee74855
commit 13ccaf1a94
2 changed files with 2 additions and 1 deletions

View File

@ -222,7 +222,8 @@ class Conformer(EncoderInterface):
x = self.layer_norm(x/self.group_num)
'''
x = self.sigmoid(self.alpha[0]) * layer_outputs[2]
x = self.layer_norm(x)
x += self.sigmoid(self.alpha[1]) * layer_outputs[5]
x = self.layer_norm(x/2)
return x, lengths