mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove 2 ConvNeXt layers.

Browse Source
This commit is contained in:
Daniel Povey 2022-12-17 21:53:46 +08:00
parent 111a0aa3c7
commit 598f52cbac
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
View File

@ -1795,7 +1795,6 @@ class Conv2dSubsampling(nn.Module):
) )
self.convnext1 = nn.Sequential(ConvNeXt(layer2_channels), self.convnext1 = nn.Sequential(ConvNeXt(layer2_channels),
ConvNeXt(layer2_channels),
ConvNeXt(layer2_channels)) ConvNeXt(layer2_channels))
self.conv2 = nn.Sequential( self.conv2 = nn.Sequential(
@ -1807,7 +1806,6 @@ class Conv2dSubsampling(nn.Module):
)) ))
self.convnext2 = nn.Sequential(ConvNeXt(layer3_channels), self.convnext2 = nn.Sequential(ConvNeXt(layer3_channels),
ConvNeXt(layer3_channels),
ConvNeXt(layer3_channels)) ConvNeXt(layer3_channels))