From 598f52cbac159d6a9064f974e990f1313ecabe4a Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 17 Dec 2022 21:53:46 +0800 Subject: [PATCH] Remove 2 ConvNeXt layers. --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index e55185051..75d24106a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1795,7 +1795,6 @@ class Conv2dSubsampling(nn.Module): ) self.convnext1 = nn.Sequential(ConvNeXt(layer2_channels), - ConvNeXt(layer2_channels), ConvNeXt(layer2_channels)) self.conv2 = nn.Sequential( @@ -1807,7 +1806,6 @@ class Conv2dSubsampling(nn.Module): )) self.convnext2 = nn.Sequential(ConvNeXt(layer3_channels), - ConvNeXt(layer3_channels), ConvNeXt(layer3_channels))