From 11cb30bf492c10b8898cbf0767f84d554255f79a Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 3 Nov 2022 17:30:36 +0800 Subject: [PATCH] Reduce bottleneck dim fo SEModule from 16 to 8. --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)