From 5d877ef25e73ad62977fa82be73a29f05afaf59a Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Fri, 17 Jun 2022 18:54:55 +0800 Subject: [PATCH] change tail padding length from right_context_length to chunk_length --- .../ASR/conv_emformer_transducer_stateless/decode.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/decode.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/decode.py index aadac2ae4..287fb94df 100755 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/decode.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/decode.py @@ -277,10 +277,10 @@ def decode_one_batch( supervisions = batch["supervisions"] feature_lens = supervisions["num_frames"].to(device) - feature_lens += params.right_context_length + feature_lens += params.chunk_length feature = torch.nn.functional.pad( feature, - pad=(0, 0, 0, params.right_context_length), + pad=(0, 0, 0, params.chunk_length), value=LOG_EPS, )