diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 2bcd7202d..33d025f86 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1434,7 +1434,7 @@ class ModifiedSEModule(nn.Module): """ def __init__(self, d_model: int, - bottleneck_dim: int = 16): + bottleneck_dim: int = 8): super().__init__() self.squeeze_proj = nn.Linear(d_model, d_model, bias=False)