From 89271fb9ea768ef4a13d4c852335978975d3c673 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 2 Feb 2023 14:07:53 +0900
Subject: [PATCH] from local

---
 .../ASR/conformer_ctc/.transformer.py.swp     | Bin 69632 -> 69632 bytes
 egs/aishell/ASR/conformer_ctc/transformer.py  |   4 +++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp
index b8c6618042c628cdb0e2e6d9f9a827e053431500..c92d73f15567b1528b1aa3359b663f8e216461ba 100644
GIT binary patch
delta 190
zcmZozz|ydQMKsAE%+puFQqO<^2m}}yPP^PruH7j5TY%AQv!LMTbG*T<3=HmU5J|Ji
zf)5s3?O<nMSPaC?Kr9Qyzksqof%rBMuL9zgK-><*2|yeV#GXLx0mSZXn;BW6S*Pc*
zGIC6w_-wa8v0h?9L26!#Mt*5YL1~HR<oVCUCi_1xlvRKNh<LGrtwMAx159x9f#(hy
E03}H<w*UYD

delta 131
zcmZozz|ydQMKsAE%+puFQqO<^2m}}yj=9`U4&NyHTY%AIv!LMTbG)-z85r8xAd)7N
z1s^OnI>64rumFgyfY=g<rGfYrP{m6iJ`ThSfOtL-cLH$-5Vx~!W@K5+Iz5k-kz;b;
Qv)z;RpO<Vt@Z3QI04WwBJ^%m!

diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py
index f8d30e403..c723fdf65 100644
--- a/egs/aishell/ASR/conformer_ctc/transformer.py
+++ b/egs/aishell/ASR/conformer_ctc/transformer.py
@@ -490,9 +490,11 @@ class TransfEncoder(nn.TransformerEncoder):
                 convert_to_nested = True
                 output = torch._nested_tensor_from_mask(output, src_key_padding_mask.logical_not(), mask_check=False)
                 src_key_padding_mask_for_layers = None
-
+        
+        outputs = []
         for mod in self.layers:
             output = mod(output, src_mask=mask, src_key_padding_mask=src_key_padding_mask_for_layers)
+            outputs.append(output)
 
         if convert_to_nested:
             output = output.to_padded_tensor(0.)