from local

This commit is contained in:
dohe0342 2023-01-04 00:11:02 +09:00
parent c52b816fab
commit 3fa5cf942c
2 changed files with 1 additions and 1 deletions

View File

@ -188,7 +188,7 @@ class Conformer(EncoderInterface):
warmup=warmup, warmup=warmup,
) # (T, N, C) ) # (T, N, C)
else: else:
x, layer_output = self.encoder( x = self.encoder(
x, x,
pos_emb, pos_emb,
src_key_padding_mask=src_key_padding_mask, src_key_padding_mask=src_key_padding_mask,