diff --git a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp index 254c0e8b4..9b65969a1 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp index 61a4d39fb..f78aaf1a8 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/conformer.py b/egs/librispeech/ASR/conformer_ctc2/conformer.py index b6983f728..69d89467a 100644 --- a/egs/librispeech/ASR/conformer_ctc2/conformer.py +++ b/egs/librispeech/ASR/conformer_ctc2/conformer.py @@ -194,9 +194,12 @@ class Conformer(Transformer): if self.group_num > 0: x = 0 - for enum, alpha in enumerate(self.alpha): - if self.learnable_alpha: x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1] - else: x += (1/self.group_num) * layer_outputs[(enum+1)*self.group_layer_num-1] + if self.learnable_alpha: + for enum, alpha in enumerate(self.alpha): + x += self.sigmoid(alpha) * layer_outputs[(enum+1)*self.group_layer_num-1] + else: + for enum in range(self.group_num): + x += (1/self.group_num) * layer_outputs[(enum+1)*self.group_layer_num-1] x = self.layer_norm(x) if self.interctc or self.interctc_condition or self.group_num > 0: