from local

2025-12-11 06:55:27 +00:00 · 2023-02-25 16:06:14 +09:00 · 2023-02-25 16:06:14 +09:00 · 9374e5d07e
commit 9374e5d07e
parent a66a1fcae0
2 changed files with 3 additions and 2 deletions
--- a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
+++ b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@ -415,9 +415,10 @@ def decode_one_batch(
        return {key: hyps}
    
    if params.method == "greedy-search":
+        memory, memory_key_padding_mask
        att_loss = model.decoder_forward(
-            encoder_memory,
-            memory_mask,
+            memory,
+            momory_key_padding_mask,
            token_ids=unsorted_token_ids,
            sos_id=graph_compiler.sos_id,
            eos_id=graph_compiler.eos_id,