From bcec8465c9ceb31c4d50f895cbdb29c9e8d92fde Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Thu, 9 Feb 2023 18:21:21 +0800 Subject: [PATCH] Update model.py --- egs/librispeech/ASR/pruned_transducer_stateless9/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless9/model.py b/egs/librispeech/ASR/pruned_transducer_stateless9/model.py index 29601807c..70ab947ad 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless9/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless9/model.py @@ -88,7 +88,7 @@ class Transducer(nn.Module): Return a tensor of shape (N, U). """ y_shift = F.pad( - y, (context_size, 0), mode="constant", value=self.decoder.blank_id + y, (context_size, 0), mode="constant", value=-1 )[:, :-context_size] mask = y_shift != y