From a5a7514ef0d7274201c423676dffb1b2bec1f5dc Mon Sep 17 00:00:00 2001 From: luomingshuang <739314837@qq.com> Date: Tue, 26 Jul 2022 16:47:38 +0800 Subject: [PATCH] change for streaming_beam_search.py --- .../ASR/pruned_transducer_stateless5/streaming_beam_search.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_beam_search.py b/egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_beam_search.py index dc4b08067..651aff6c9 100644 --- a/egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_beam_search.py +++ b/egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_beam_search.py @@ -152,8 +152,6 @@ def modified_beam_search( index=hyps_shape.row_ids(1).to(torch.int64), ) # (num_hyps, encoder_out_dim) - print(decoder_out.shape) - print(current_encoder_out.shape) logits = model.joiner( current_encoder_out, decoder_out.unsqueeze(1),