mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix padding issues (#1303)

Browse Source
This commit is contained in:
Wen Ding 2023-10-11 16:58:00 +08:00 committed by GitHub
parent 16a2748d6c
commit 2b3c5d799f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py
View File

@ -158,7 +158,7 @@ class Conformer(EncoderInterface):
if not is_jit_tracing(): if not is_jit_tracing():
assert x.size(0) == lengths.max().item() assert x.size(0) == lengths.max().item()
src_key_padding_mask = make_pad_mask(lengths) src_key_padding_mask = make_pad_mask(lengths, x.size(0))
if self.dynamic_chunk_training: if self.dynamic_chunk_training:
assert ( assert (

2
egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
View File

@ -281,7 +281,7 @@ class Zipformer(EncoderInterface):
lengths = (x_lens - 7) >> 1 lengths = (x_lens - 7) >> 1
assert x.size(0) == lengths.max().item(), (x.shape, lengths, lengths.max()) assert x.size(0) == lengths.max().item(), (x.shape, lengths, lengths.max())
mask = make_pad_mask(lengths) mask = make_pad_mask(lengths, x.size(0))
outputs = [] outputs = []
feature_masks = self.get_feature_masks(x) feature_masks = self.get_feature_masks(x)