mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
1a91afa118
commit
4336856a25
Binary file not shown.
@ -150,6 +150,10 @@ class Conformer(Transformer):
|
||||
) # (T, N, C)
|
||||
|
||||
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]
|
||||
x = self.layer_norm(x/self.group_num)
|
||||
# x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user