Make decoder group size equal to 4.
This commit is contained in:
parent
ff03ec88a5
commit
bba454a349
@ -69,7 +69,7 @@ class Decoder(nn.Module):
|
|||||||
out_channels=decoder_dim,
|
out_channels=decoder_dim,
|
||||||
kernel_size=context_size,
|
kernel_size=context_size,
|
||||||
padding=0,
|
padding=0,
|
||||||
groups=decoder_dim,
|
groups=decoder_dim//4, # group size == 4
|
||||||
bias=False,
|
bias=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user