From 9460ff6b38b954c2c67e82065a72a9235d927f86 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 7 Jan 2023 00:30:29 +0900 Subject: [PATCH] from local --- .../.conformer.py.swp | Bin 94208 -> 94208 bytes .../conformer.py | 5 +++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp index 86ff639e67528345dba5b73cc1bf9bf338f9c961..88b4a9edaf2712a064c801e8a86e6f5dd7823e20 100644 GIT binary patch delta 320 zcmZp8z}oPDRV>LM%+puFQqO<^2m}}y8trx@*FN1SCMCca-)tzb-B5t>vOO;wP;wq4 zL`D2`!2rhd^;0<*7%G6+5{N~C_$507!wVq34#a&xoCL&SK^ lr3LZDnN_KpnhXk1fF`40t6*pZmz@4TlTmW}wg^TqaR6#PKg9q5 delta 271 zcmZp8z}oPDRV>LM%+puFQqO<^2m}}ywp#5-PI|aeOiF+;uGvsvyP*K%WqV#5pyUNc zh>E!Bf&q-@>*sPXFk}O+yTT@K%5K2azM-j#CzEo80G?T zHxSzb@q1PVhC@KS5r~_B*c6CgurM%e0OBP;Tn)t8KFk)x{ zGW>wp7l>7XSOAE>FflN!2I6iY_6A~hAbtdNH^}W5rY{s^w4XGIXZxx^#yxB-hDMs2 PlM`K}wr`7I^b!XEN2D;> diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py index d22ed5962..c673421a7 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py @@ -134,8 +134,9 @@ class Conformer(EncoderInterface): ), ) self._init_state: List[torch.Tensor] = [torch.empty(0)] - - self.alpha = nn.Parameter(torch.rand(12)) + + self.group_size = 12 + self.alpha = nn.Parameter(torch.rand(self.group_size)) self.sigmoid = nn.Sigmoid() self.layer_norm = nn.LayerNorm(512)