diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 66d5db892..700f64963 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1459,7 +1459,7 @@ class NonlinAttentionModule(nn.Module): """ def __init__( - self, channels: int, ratio: int = 8, + self, channels: int, ratio: int = 2, ) -> None: super().__init__()