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

[agnieszkajust]

I get "RuntimeError: The size of tensor a (335) must match the size of tensor b (334) at non-singleton dimension 3" during testing on my own data. How to solve it?

[chosj95]

Hi,

I think you tested on the images of which size are not multiple of 4.
Because of the stride in our model, you should use the input image having sizes of multiple of 4.
If necessary, you should pad an input image and remove the padded area.

Which dataset did you test on?

[agnieszkajust]

I tried to test on RealBlur - http://cg.postech.ac.kr/research/realblur/ .

[chosj95]

Since the RealBlur dataset contains the images of which size are not multiple of 4, we also pad the input image and remove the padded area from the output image.

[agnieszkajust]

Ok, I'll do it. Thanks.
Will be available a code to training MIMO-UNet++ soon?

[usutdzxych]

no results