From 3554ed32dbe56dc3b4af12c0d3f980b0e0c286be Mon Sep 17 00:00:00 2001 From: yifanyang Date: Mon, 6 Feb 2023 18:47:36 +0800 Subject: [PATCH] Fix for style check --- .../ASR/pruned_transducer_stateless9/beam_search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py index c5ae0669e..605d15fa4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py @@ -720,9 +720,9 @@ def greedy_search_batch( k = torch.sum( ( - c[:, : context_size] + c[:, :context_size] == c[:, context_size : context_size + 1].expand_as( - c[:, : context_size] + c[:, :context_size] ) ).int(), dim=1,