naoto0804 / pytorch-inpainting-with-partial-conv

Unofficial pytorch implementation of 'Image Inpainting for Irregular Holes Using Partial Convolutions' [Liu+, ECCV2018]

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Problem while using net.py

WajdiMuh opened this issue · 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