diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index ab275977e..bfeb00ac9 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 79fd6bc8e..214a8fbec 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -181,7 +181,12 @@ class Conformer(Transformer): mask = mask.to(x.device) if mask is not None else None x, layer_outputs = self.encoder( - x, pos_emb, src_key_padding_mask=mask, warmup=warmup, condition_layer=self.condition_layer, + x, + pos_emb, + src_key_padding_mask=mask, + warmup=warmup, + condition_layer=self.condition_layer, + ctc_output=self.ctc_output, ) # (S, N, C) if self.group_num != 0: