naoto0804/pytorch-inpainting-with-partial-conv

Problem while using net.py

WajdiMuh opened this issue · 1 comments

when i tried to link your model and loss function into a project of mine i encountered thus issue
RuntimeError: Sizes of tensors must match except in dimension 1. Got 10 and 9 in dimension 2 (The offending index is 1)
line h = torch.cat([h, h_dict[enc_h_key]], dim=1)
the sizes of the tensors in order were
torch.Size([32, 512, 10, 10])
torch.Size([32, 512, 9, 9])
can you help me fix it
thanks in advance

turns it it is a problem with the image size im not using it on the places dataset