mirrors/icefall
Archived
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

change for streaming_beam_search.py

Browse Source
This commit is contained in:
luomingshuang 2022-07-26 16:47:38 +08:00
parent 4bf839227c
commit a5a7514ef0
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_beam_search.py
View File

@ -152,8 +152,6 @@ def modified_beam_search(
index=hyps_shape.row_ids(1).to(torch.int64), index=hyps_shape.row_ids(1).to(torch.int64),
) # (num_hyps, encoder_out_dim) ) # (num_hyps, encoder_out_dim)
print(decoder_out.shape)
print(current_encoder_out.shape)
logits = model.joiner( logits = model.joiner(
current_encoder_out, current_encoder_out,
decoder_out.unsqueeze(1), decoder_out.unsqueeze(1),