from local

2025-12-11 06:55:27 +00:00 · 2023-02-25 15:39:08 +09:00 · 2023-02-25 15:39:08 +09:00 · 9446c425db
commit 9446c425db
parent 8a1a0cb3e9
2 changed files with 2 additions and 0 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
@ -342,6 +342,8 @@ def decode_one_batch(
    supervisions = batch["supervisions"]

    nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
+    if type(nnet_output) == tuple:
+        nnet_output = nnet_output[0]

    supervision_segments = torch.stack(
        (