From 1005ff35bad1881396d50590bd643c5c26448d5b Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 28 Sep 2022 13:57:26 +0800 Subject: [PATCH] Fix w.r.t. uneven upsampling --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 312ffe0e6..79a6ebefb 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -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