diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index 787cab876..abc006402 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/conformer.py b/egs/tedlium2/ASR/conformer_ctc3/conformer.py index 07db5498b..4e760e1ec 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -180,6 +180,7 @@ 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/self.group_num) + return x, mask