mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Bug fix
This commit is contained in:
parent
9023fe7151
commit
bc9fbe2579
@ -508,7 +508,7 @@ class ConformerEncoder(nn.Module):
|
||||
for i, mod in enumerate(self.layers):
|
||||
if i in layers_to_drop:
|
||||
continue
|
||||
next_output, attn_scores = mod(
|
||||
output, attn_scores = mod(
|
||||
output,
|
||||
pos_emb,
|
||||
attn_scores,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user