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