convnext1 kernel size 5, 5 to 5, 7

This commit is contained in:
Daniel Povey 2023-01-02 14:17:51 +08:00
parent 3a5b3f640d
commit a61bd01e5b

View File

@ -1842,8 +1842,8 @@ class Conv2dSubsampling(nn.Module):
SwooshR(), SwooshR(),
) )
self.convnext1 = nn.Sequential(ConvNeXt(layer2_channels, kernel_size=(5, 5)), self.convnext1 = nn.Sequential(ConvNeXt(layer2_channels, kernel_size=(5, 7)),
ConvNeXt(layer2_channels, kernel_size=(5, 5)), ConvNeXt(layer2_channels, kernel_size=(5, 7)),
BasicNorm(layer2_channels, BasicNorm(layer2_channels,
channel_dim=1)) channel_dim=1))