From a424a7388122717adcf05cb9d9eae9bfd0ecd61a Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 17 Dec 2022 21:58:59 +0800 Subject: [PATCH] Increase ratio of convnext from 3 to 4. --- 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 6e938837d..197362c9d 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1673,7 +1673,7 @@ class ConvNeXt(nn.Module): """ def __init__(self, channels: int, - hidden_ratio: int = 3, + hidden_ratio: int = 4, layerdrop_prob: FloatLike = None): super().__init__() kernel_size = 7