pytorch_misc
pytorch_misc copied to clipboard
unet_demo.py: fix nn.Conv2d argument order
The padding
and stride
arguments are swapped in BaseConv
. Note that this does not fix that the architecture does not work with stride != 1
.