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

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

diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp
index 0401582c7bf1d0803c8aee1ef2cfe75094dece29..05e41821ac3d1f31f5fc17b6eb199ada996d7f1a 100644
GIT binary patch
delta 214
zcmZozz|ydQMJ&l6%+puFQqO<^2m}}yRy*HL4rAUZwo8Cfa`Qz2<NdrHTnr2x+z?sG
z$$}T0Raf#bFw_8XC=go#@qeJ~O&~rE#HWDR4Tw25S6=Ain#_OOYO>UwX_Hrbipk|B
z7H7vNXQU=)+aj0@3Q#av`o4^)ZE9X(Qch}oUTSekYD#=bYF=@E(d0&NVV3-&T-(X#
nJXKk95-U@SCLeq)Jo&S)+T__i+>`g*k(k`ST4M9tJHo~Q5}reW

delta 176
zcmZozz|ydQMJ&l6%+puFQqO<^2m}}y+MI7E%QI~h+a<s#y!oPl@qXS~ZU%;@Kn#)<
zo-BC5S#<>u1H*D4Rt92_%o8Af48(0foD0MrKrFbq@<JCETXB9-tb*<2hpUAr3*6S7
z%z0Z?LIDDll#~Jzi;ERXGEx;X^9o8!CUe}j1Y(uRY`4!&p4G!MdG8$wM$O6nt0gwS
Iy(4T40A83gasU7T

diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py
index 58de71c90..d85f7d7d8 100644
--- a/egs/aishell/ASR/conformer_ctc/transformer.py
+++ b/egs/aishell/ASR/conformer_ctc/transformer.py
@@ -401,7 +401,13 @@ class TransfEncoder(nn.TransformerEncoder):
     __constants__ = ['norm']
 
     def __init__(self, encoder_layer, num_layers, norm=None, enable_nested_tensor=True, mask_check=True):
-        super(TransfEncoder, self).__init__()
+        super(TransfEncoder, self).__init__(
+                encoder_layer=encoder_layer,
+                num_layers=num_layers,
+                norm=norm,
+                enable_nested_tensor=enable_nested_tensor,
+                mask_check=mask_check
+            )
 
     def forward(self, src: torch.Tensor, mask: Optional[torch.Tensor] = None, src_key_padding_mask: Optional[torch.Tensor] = None):
         """Pass the input through the encoder layers in turn.