StyleSwin
StyleSwin copied to clipboard
Size mismatch
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
Hi, just set --size 512 in the command line to generate images of size 512x512.