avalonstrel/GatedConvolution_pytorch

about the package version

wdrink opened this issue · 1 comments

Hi, thanks for your work in implementing the code with pytorch. I encountered an error that the netD will crash if using 64*64 as size of input img and mask:

RuntimeError: Calculated padded input size per channel: (3 x 3). Kernel size: (4 x 4). Kernel size can't be greater
than actual input size

The same code works well when I set the size to 128128 or 256256.

请问你解决了嘛?遇到同样问题 求教 多谢~~

I solved. To modify these 2 lines to 256*256.

TRAIN_SIZES = ((64,64),(128,128),(256,256))
SIZES_TAGS = ("64x64", "128x128", "256x256")

FYI~