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
72679fa259
commit
6cc8b6fcd1
Binary file not shown.
@ -195,7 +195,7 @@ class Conformer(EncoderInterface):
|
||||
num_left_chunks=self.num_left_chunks,
|
||||
device=x.device,
|
||||
)
|
||||
x, layer_output = self.encoder(
|
||||
x, layer_outputs = self.encoder(
|
||||
x,
|
||||
pos_emb,
|
||||
mask=mask,
|
||||
@ -203,7 +203,7 @@ class Conformer(EncoderInterface):
|
||||
warmup=warmup,
|
||||
) # (T, N, C)
|
||||
else:
|
||||
x, layer_output = self.encoder(
|
||||
x, layer_outputs = self.encoder(
|
||||
x,
|
||||
pos_emb,
|
||||
src_key_padding_mask=src_key_padding_mask,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user