from local

This commit is contained in:
dohe0342 2023-01-21 15:31:20 +09:00
parent f07709d3c1
commit 6a64b688f9
2 changed files with 1 additions and 1 deletions

View File

@ -137,7 +137,7 @@ class Conformer(Transformer):
# Caution: We assume the subsampling factor is 4!
x = self.encoder(
x, layer_outputs = self.encoder(
x, pos_emb, src_key_padding_mask=mask, warmup=warmup
) # (T, N, C)