diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index 6c7dc8019..40b834e45 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 d84e1bfee..3b3176dc9 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -171,7 +171,7 @@ class Conformer(Transformer): mask = encoder_padding_mask(x.size(0), supervisions) mask = mask.to(x.device) if mask is not None else None - x = self.encoder( + x, layer_outputs = self.encoder( x, pos_emb, src_key_padding_mask=mask, warmup=warmup ) # (S, N, C)