from local

This commit is contained in:
dohe0342 2023-01-09 19:25:11 +09:00
parent 1c11c19869
commit 04800893e1
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 = self.encoder( x, layer_outputs = self.encoder(
x, x,
pos_emb, pos_emb,
src_key_padding_mask=src_key_padding_mask, src_key_padding_mask=src_key_padding_mask,