generative-inpainting-pytorch icon indicating copy to clipboard operation
generative-inpainting-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon daa233

Problem in contextual attention module

Open XiaoqiangZhou opened this issue 5 years ago • 1 comments

Thanks for your reimplementation.

When I run the test_contextual_attention function in the network.py. Error occured. RuntimeError: Given transposed=1, weight of size 4096 3 4 4, expected input[1, 4032, 166, 250] to have 4096 channels, but got 4032 channels instead In my setting, --imagA is the bnw_butterfly.png and --imageB is the bike.jpg for the offical CA repo, in the folder of examples/style_transfer.

Could you please help me with it? Thanks.

Best,

XiaoqiangZhou avatar Nov 08 '19 09:11 XiaoqiangZhou

It is probably because the contextual attention module here doesn't support arbitrary image size, refers to #14.

daa233 avatar Nov 29 '19 12:11 daa233