diff --git a/egs/librispeech/ASR/.gtrans.sh.swp b/egs/librispeech/ASR/.gtrans.sh.swp
index 17a6f69d1..abf0029b8 100644
Binary files a/egs/librispeech/ASR/.gtrans.sh.swp and b/egs/librispeech/ASR/.gtrans.sh.swp differ
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 264a91159..11ea0a55c 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
index 5b9f675b9..6ea83883a 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
@@ -127,7 +127,7 @@ class Conformer(EncoderInterface):
 
         self.group_num = group_num
         self.group_layer_num = int(self.encoder_layers // self.group_num)
-        self.alpha = nn.Parameter(torch.rand(self.group_size))
+        self.alpha = nn.Parameter(torch.rand(self.group_num))
         self.sigmoid = nn.Sigmoid()
         self.layer_norm = nn.LayerNorm(d_model)