about the package version

[wdrink]

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.

[Francoforever]

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

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

GatedConvolution_pytorch/test_images.py

Lines 33 to 34 in 22455da

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

FYI~