From 753269668ac4985e947da07cc8e512c251036728 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 29 Nov 2022 15:38:13 +0800 Subject: [PATCH] Change ratio in NonlinAttentionModule from 8 to 2 --- 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 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__()