mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix w.r.t. uneven upsampling

Browse Source
This commit is contained in:
Daniel Povey 2022-09-28 13:57:26 +08:00
parent 10a3061025
commit 1005ff35ba
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -456,6 +456,8 @@ class DownsampledConformerEncoder(nn.Module):
src, src_key_padding_mask=mask, warmup=warmup
)
src = self.upsample(src)
# remove any extra frames that are not a multiple of downsample_factor
src = src[:src_orig.shape[0]]
return src