From 5b10310bd19e9abbdc1fa22c1ff6984cdafcde47 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 11 Nov 2021 15:42:30 +0800 Subject: [PATCH] Handle empty lattices in attention decoder rescoring. --- icefall/decode.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/icefall/decode.py b/icefall/decode.py index aef37cb69..4c2a8e01b 100644 --- a/icefall/decode.py +++ b/icefall/decode.py @@ -856,6 +856,10 @@ def rescore_with_attention_decoder( tokens = tokens.remove_values_leq(0) token_ids = tokens.tolist() + if len(token_ids) == 0: + print("Warning: rescore_with_attention_decoder(): empty token-ids") + return None + nll = model.decoder_nll( memory=expanded_memory, memory_key_padding_mask=expanded_memory_key_padding_mask,