Problem in contextual attention module

[XiaoqiangZhou]

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,

[daa233]

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