Error while concat decoder_4_f and encoder_3

[Sivarajakani]

I have loaded the model weight successfully. While doing inference, I'm facing an error in the Focus decoder while concatenating the result of decoder_4_f and encoder_3. I'm using r34 as a backend and TT as a rosta.

d4_f shape - [1, 256, 20, 28]
e3 shape - [1, 256, 20, 27]

Error:

File "gfm.py", line 411
d3_f = self.decoder3_f(torch.cat((d4_f, e3), 1))
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 28 but got size 27 for tensor number 1 in the list.

Versions:
OS - Ubuntu 20
python - 3.9.9
torch - 1.10.1
torchvision - 0.11.2

[Sivarajakani]

I fixed the issue. I made mistake in resizing the image. While resizing the image, the new size should be divisible by 32.