diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index 9fd22fe37..3ec2ecd0c 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/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index 7744b5032..119c5ac49 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/conformer.py b/egs/tedlium2/ASR/conformer_ctc3/conformer.py index bcfffdd19..ba1a5ad4c 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -144,6 +144,8 @@ class Conformer(Transformer): self.interctc_condition = interctc_condition if self.interctc_condition: self.condition_layer = ScaledLinear(500, d_model) + else: + self.condition_layer = None def run_encoder( self, @@ -179,7 +181,7 @@ 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 + x, pos_emb, src_key_padding_mask=mask, warmup=warmup, ) # (S, N, C) if self.group_num != 0: