From 06b583ee8494810832a8ea5e4c938474058bb335 Mon Sep 17 00:00:00 2001 From: yifanyang Date: Mon, 6 Feb 2023 18:51:43 +0800 Subject: [PATCH] Fix for style check --- egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py index 605d15fa4..341fc6611 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py @@ -672,7 +672,7 @@ def greedy_search_batch( timestamps = [[] for _ in range(N)] decoder_input = torch.tensor(hyps, device=device, dtype=torch.int64)[ - :, -context_size: + :, -context_size: ] # (N, context_size) k = torch.zeros(N, 1, device=device, dtype=torch.int64)