from local

This commit is contained in:
dohe0342 2023-02-14 02:49:03 +09:00
parent 3038c3f5f3
commit 239e697170
2 changed files with 1 additions and 1 deletions

View File

@ -171,7 +171,7 @@ class Conformer(Transformer):
mask = encoder_padding_mask(x.size(0), supervisions)
mask = mask.to(x.device) if mask is not None else None
x = self.encoder(
x, layer_outputs = self.encoder(
x, pos_emb, src_key_padding_mask=mask, warmup=warmup
) # (S, N, C)