photo2cartoon icon indicating copy to clipboard operation
photo2cartoon copied to clipboard

Training on input size (386*386) given an error

Open pradyumnjain opened this issue 2 years ago • 1 comments

The training works fine for 128 and 256 input sizes but doesn't work when the input image is not in the power of 2. I received the following error when training with 386*386

RuntimeError: The size of tensor a (193) must match the size of tensor b (192) at non-singleton dimension 3

pradyumnjain avatar Jun 17 '22 10:06 pradyumnjain

Due to several down-sampling layers, the input size should be divisible by 2 many times. Please try 384*384.

hao-qiang avatar Jul 04 '22 08:07 hao-qiang