Update conformer.py

非dynamic_chunk训练时,传入参数修正
This commit is contained in:
l2009312042 2023-08-04 19:06:12 +08:00 committed by GitHub
parent 1ee251c8b3
commit c196fb4438
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -230,7 +230,7 @@ class Conformer(Transformer):
x, pos_emb, mask=mask, src_key_padding_mask=src_key_padding_mask
) # (T, B, F)
else:
x = self.encoder(x, pos_emb, src_key_padding_mask=mask) # (T, B, F)
x = self.encoder(x, pos_emb, src_key_padding_mask=src_key_padding_mask) # (T, B, F)
if self.normalize_before:
x = self.after_norm(x)