diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index 1244cd7be..6174319e3 100644 Binary files a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp and b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp differ diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py index eb2141b86..c7fa420c2 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -224,7 +224,7 @@ class Transformer(nn.Module): x = x.permute(1, 0, 2) # (N, T, C) -> (T, N, C) mask = encoder_padding_mask(x.size(0), supervisions) mask = mask.to(x.device) if mask is not None else None - x, layer_outputs= self.encoder(x, src_key_padding_mask=mask) # (T, N, C) + x, layer_outputs = self.encoder(x, src_key_padding_mask=mask) # (T, N, C) return x, mask diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp deleted file mode 100644 index bd733e9b8..000000000 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp and /dev/null differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py index d4d5d38ee..c8e252fe5 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py @@ -205,7 +205,7 @@ class Conformer(EncoderInterface): x = 0 for enum, alpha in enumerate(self.alpha): - 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/self.group_num)