mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Bug fix

Browse Source
This commit is contained in:
Daniel Povey 2022-10-03 13:23:26 +08:00
parent c20fc3be14
commit a0a1874415
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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