From b67c078fb44db340ecdbed7aa638f1a18d889a83 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 2 Feb 2023 14:15:21 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc/.transformer.py.swp | Bin 57344 -> 61440 bytes egs/aishell/ASR/conformer_ctc/transformer.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index 1554aba78f277d4075054e026c3cfb7b7c5203d1..1244cd7beae1fc6417a9c44b71cf5efac30dcee9 100644 GIT binary patch delta 790 zcmY+>Ur1A77{~G7*;%%!XU=Rbx@nSiQ6okdNid{LqC*yRp%z3nOCl_$wn58c8U#T~ z>FK7((N*?G=+6O-@Itx>i>RP7Xfy-~jc%m747Bgan;!V^K8N%2zUMr@o(Z#O+6=YE zn(h6O9b03fg3{cc{E5QT&{xxFa_7xe<%Z>$!Lvrl=?sP|1I100{cq8CHxoQn@hsxq zX{|4luA_KH$f-Quk}W>f;{r0M!#30)jC`r|9AmhL)7XnRs*v|fL+Habq;UiX5QBkl zK4}sz?%)P4qr<1HVl);CE2Y=CggQh~1p{*>(k!OXheK$@I{a864PpRCu?16J=`%*r zk5(iQfQ4~W>cR#yDt+=u4{#rC*aa_U4Cx(Q+(gcBhZ0Zx&f=Iaa6TG2lgPAn9ZRND z9cgE{y>@M6x&D8ZI_D0zA5LWxC#9F%?BR=x!?%~SS*qagFpCk~#UQT0#wKjUYJ4Hs zQ9ML1Y?NWal73?bZ!nG=lGums7GtiaDaB&BG>-yu=td)I@sqqqa2vfyq8UwSzyf){ z!V6qQ3xp*yev4aZz;4vw53zp0dtY3-NYjnYm?qN!)FX^F_{d})qYDQSM>RYpuHl{Z RIuo5_${5iNPE-1Fk;T@^2txfGwbXRYyATqr}NI`y1Mx#pA7 z#QRXax~4nI$Z2V3!|3y>#zN@e&N{bzligkHpKAX{X{To~S*S27V;4KHF@X_8(F}^* z;2JLSn8P$8Xa;zX5)M(s3Q|Zw$D3c|9yOfd82i}tOF8WwqN6S;tgb<7}y7CAklhQG{C YG{+q<{9 diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py index 1012b0921..eb2141b86 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -224,7 +224,7 @@ class Transformer(nn.Module): x = x.permute(1, 0, 2) # (N, T, C) -> (T, N, C) mask = encoder_padding_mask(x.size(0), supervisions) mask = mask.to(x.device) if mask is not None else None - x = self.encoder(x, src_key_padding_mask=mask) # (T, N, C) + x, layer_outputs= self.encoder(x, src_key_padding_mask=mask) # (T, N, C) return x, mask