mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Avoid error in ddp by using last module'sc scores
This commit is contained in:
parent
0f85a3c2e5
commit
c16f795962
@ -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
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user