from local

This commit is contained in:
dohe0342 2023-02-16 11:00:55 +09:00
parent bbb8b8d1ba
commit 90ef4d89e6
3 changed files with 1 additions and 1 deletions

View File

@ -195,7 +195,7 @@ class Conformer(Transformer):
x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1] x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1]
x = self.layer_norm(x) x = self.layer_norm(x)
if self.interctc or self.group_num > 0: if self.interctc or self.interctc_condition or self.group_num > 0:
return (x, layer_outputs), mask return (x, layer_outputs), mask
else: else:
return x, mask return x, mask