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
4efa6c14b0
commit
28c3ef7c24
Binary file not shown.
@ -183,8 +183,11 @@ class Conformer(Transformer):
|
||||
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)
|
||||
|
||||
return x, mask
|
||||
|
||||
if self.interctc:
|
||||
return layer_outputs, mask
|
||||
else:
|
||||
return x, mask
|
||||
|
||||
|
||||
class ConformerEncoderLayer(nn.Module):
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user