From 834114f1c28334543b3af0fe4a995ba847448d9c Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 22 Dec 2021 23:49:31 +0800 Subject: [PATCH] Update decoder.py --- egs/librispeech/ASR/transducer_stateless/decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/transducer_stateless/decoder.py b/egs/librispeech/ASR/transducer_stateless/decoder.py index f49f8c90b..cedbc937e 100644 --- a/egs/librispeech/ASR/transducer_stateless/decoder.py +++ b/egs/librispeech/ASR/transducer_stateless/decoder.py @@ -76,8 +76,8 @@ class Decoder(nn.Module): y: A 2-D tensor of shape (N, U) with blank prepended. need_pad: - True to left padding the input. Should be True during training. - False to not pad the input. Used only during inference. + True to left pad the input. Should be True during training. + False to not pad the input. Should be False during inference. Returns: Return a tensor of shape (N, U, embedding_dim). """