Make decoder group size equal to 4.

This commit is contained in:
Daniel Povey 2022-10-29 17:08:19 +08:00
parent ff03ec88a5
commit bba454a349

View File

@ -69,7 +69,7 @@ class Decoder(nn.Module):
out_channels=decoder_dim,
kernel_size=context_size,
padding=0,
groups=decoder_dim,
groups=decoder_dim//4, # group size == 4
bias=False,
)