mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
3038c3f5f3
commit
239e697170
Binary file not shown.
@ -171,7 +171,7 @@ class Conformer(Transformer):
|
||||
mask = encoder_padding_mask(x.size(0), supervisions)
|
||||
mask = mask.to(x.device) if mask is not None else None
|
||||
|
||||
x = self.encoder(
|
||||
x, layer_outputs = self.encoder(
|
||||
x, pos_emb, src_key_padding_mask=mask, warmup=warmup
|
||||
) # (S, N, C)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user