score_sde_pytorch icon indicating copy to clipboard operation
score_sde_pytorch copied to clipboard

padding of first convolution set to dilation in residualblock

Open ParsaEsmati opened this issue 1 year ago • 0 comments

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.

ParsaEsmati avatar Jul 14 '24 12:07 ParsaEsmati