score_sde_pytorch
score_sde_pytorch copied to clipboard
padding of first convolution set to dilation in residualblock
In the case where downsampling is true and dilation>1 there are two convolutions both of which are downsampling by the number of dilation. However the shortcut is only downsampled with one dilation which creates a missmatch between shortcut and output when downsampling is true and dilation>1.
This fix sets the padding=dilation for the first convolution of the ResidualBlock in the case with downsampling 'true' and dilation>1.