from local

This commit is contained in:
dohe0342 2023-01-09 11:25:37 +09:00
parent 72679fa259
commit 6cc8b6fcd1
2 changed files with 2 additions and 2 deletions

View File

@ -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,