from local

2025-12-11 06:55:27 +00:00 · 2023-01-21 15:30:57 +09:00 · 2023-01-21 15:30:57 +09:00 · c681ce6c65
commit c681ce6c65
parent 8b8bf8250b
2 changed files with 4 additions and 0 deletions
--- a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp
+++ b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp
--- a/egs/librispeech/ASR/conformer_ctc2/conformer.py
+++ b/egs/librispeech/ASR/conformer_ctc2/conformer.py
@ -328,6 +328,8 @@ class ConformerEncoder(nn.Module):
        """
        output = src

+        layer_outputs = []
+
        for i, mod in enumerate(self.layers):
            output = mod(
                output,
@ -337,6 +339,8 @@ class ConformerEncoder(nn.Module):
                warmup=warmup,
            )

+            layer_outputs.append(output)
+
        return output