From c16f79596203fd5b77eb0dcbb5f495862a7138f8 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 22 Sep 2022 18:52:16 +0800 Subject: [PATCH] Avoid error in ddp by using last module'sc scores --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 ++ 1 file changed, 2 insertions(+) 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