mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle empty lattices in attention decoder rescoring.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-11-11 15:42:30 +08:00
parent 8d679c3e74
commit 5b10310bd1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
icefall/decode.py
View File

@ -856,6 +856,10 @@ def rescore_with_attention_decoder(
tokens = tokens.remove_values_leq(0) tokens = tokens.remove_values_leq(0)
token_ids = tokens.tolist() token_ids = tokens.tolist()
if len(token_ids) == 0:
print("Warning: rescore_with_attention_decoder(): empty token-ids")
return None
nll = model.decoder_nll( nll = model.decoder_nll(
memory=expanded_memory, memory=expanded_memory,
memory_key_padding_mask=expanded_memory_key_padding_mask, memory_key_padding_mask=expanded_memory_key_padding_mask,