From c9a35bcc73df83bbfeada15cfc3982420543681e Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sun, 8 Jan 2023 18:50:17 +0900 Subject: [PATCH] from local --- .../.conformer.py.swp | Bin 94208 -> 94208 bytes .../conformer.py | 1 + 2 files changed, 1 insertion(+) 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 915de9be4bee85d06f631e6fe857c9e369707740..a2e17c052797d2c3d5ea8570e4c0e7873ce9634e 100644 GIT binary patch delta 165 zcmZp8z}oPDRV>LM%+puFQqO<^2m}}yZcN;jY#_Z+OiF-Jf3u;$<@>z9*%=rvb3kPE zCkuYq%$LT=z@QGq%s_mZWAmjCj2x5Yel9c1NvupQiq9*}Rj^ga%q!6-PR&WvOU+Bp yPXP*pC5sjG^%cOv=|%aa1@Xn1RjHfT|J=yNcw%cJV>&MLM%+puFQqO<^2m}}y7Eauid`@bkn3Mpc-eyCA%lCP`I2ah@I3con zlLbF)<|_j7{D2rF>dm?N(g#M4>GN3`y(eG(v262^pL5t4FKkU@Oy}iPfC7f?7kL@= FlK>EkATR&` diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py index 28780e0c7..643858af0 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py @@ -126,6 +126,7 @@ class Conformer(EncoderInterface): self._init_state: List[torch.Tensor] = [torch.empty(0)] self.group_size = group_size + self.group_layer_num = int(self.encoder_layers // self.group_size) self.alpha = nn.Parameter(torch.rand(self.group_size)) self.sigmoid = nn.Sigmoid() self.layer_norm = nn.LayerNorm(d_model)