StyleSwin icon indicating copy to clipboard operation
StyleSwin copied to clipboard

Size mismatch

Open Nd-sole opened this issue 2 years ago • 1 comments

I am using generator with image size: 512, size: 256, style_dim: 512, channel_multiplier: 2. I am getting error:

Can you let me know what should be size config to have the network for image size 512 x 512?

generator.py", line 252, in forward
out = gamma * out + beta
RuntimeError: The size of tensor a (1024) must match the size of tensor b (512) at non-singleton dimension 2

Nd-sole avatar May 23 '23 15:05 Nd-sole

Hi, just set --size 512 in the command line to generate images of size 512x512.

ForeverFancy avatar Jul 08 '23 13:07 ForeverFancy