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),