from local

This commit is contained in:
dohe0342 2023-02-14 12:57:52 +09:00
parent 4efa6c14b0
commit 28c3ef7c24
2 changed files with 5 additions and 2 deletions

View File

@ -184,7 +184,10 @@ class Conformer(Transformer):
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):