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):
|
for i, mod in enumerate(self.layers):
|
||||||
if i in layers_to_drop:
|
if i in layers_to_drop:
|
||||||
continue
|
continue
|
||||||
next_output, attn_scores = mod(
|
output, attn_scores = mod(
|
||||||
output,
|
output,
|
||||||
pos_emb,
|
pos_emb,
|
||||||
attn_scores,
|
attn_scores,
|
||||||
|
|||||||
Reference in New Issue
Block a user