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
b6a0ae8b61
commit
01fbc6c4b9
Binary file not shown.
@ -181,7 +181,7 @@ class Conformer(Transformer):
|
|||||||
mask = mask.to(x.device) if mask is not None else None
|
mask = mask.to(x.device) if mask is not None else None
|
||||||
|
|
||||||
x, layer_outputs = self.encoder(
|
x, layer_outputs = self.encoder(
|
||||||
x, pos_emb, src_key_padding_mask=mask, warmup=warmup,
|
x, pos_emb, src_key_padding_mask=mask, warmup=warmup, condition_layer=self.condition_layer,
|
||||||
) # (S, N, C)
|
) # (S, N, C)
|
||||||
|
|
||||||
if self.group_num != 0:
|
if self.group_num != 0:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user