From a0a187441534f914eb5c1360852d5955246b3b39 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 3 Oct 2022 13:23:26 +0800 Subject: [PATCH] Bug fix --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 2edff7889..d8b1261a6 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -391,10 +391,12 @@ class ConformerEncoder(nn.Module): ) if i in self.aux_layers: outputs.append(output) + if i == num_layers - 1: + final_output = output output = self.combiner(outputs) - output = output + 0.0 * attn_scores.sum() # just ensure attn_scores is used in backprop + output = final_output + 0.0 * attn_scores.sum() # just ensure attn_scores is used in backprop return output