Fix padding bug

This commit is contained in:
Daniel Povey 2022-03-30 11:18:04 +08:00
parent ca6337b78a
commit 21a099b110

View File

@ -974,7 +974,7 @@ class Conv2dSubsampling(nn.Module):
self.conv = nn.Sequential( self.conv = nn.Sequential(
ScaledConv2d( ScaledConv2d(
in_channels=1, out_channels=layer1_channels, in_channels=1, out_channels=layer1_channels,
kernel_size=3, kernel_size=3, padding=1,
), ),
ActivationBalancer(channel_dim=1), ActivationBalancer(channel_dim=1),
DoubleSwish(), DoubleSwish(),