mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid error in ddp by using last module'sc scores

Browse Source
This commit is contained in:
Daniel Povey 2022-09-22 18:52:16 +08:00
parent 0f85a3c2e5
commit c16f795962
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
View File

@ -358,6 +358,8 @@ class ConformerEncoder(nn.Module):
output = self.combiner(outputs) output = self.combiner(outputs)
output = output + 0.0 * attn_scores.sum() # just ensure attn_scores is used in backprop
return output return output