diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 92f3f2dc7..385bfda36 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -358,6 +358,8 @@ class ConformerEncoder(nn.Module): output = self.combiner(outputs) + output = output + 0.0 * attn_scores.sum() # just ensure attn_scores is used in backprop + return output