[Question] About the upscaler

[LouSparfell]

Hello lucidrains,

Thanks for your code!

I was wondering, is it possible to upscale an image from 1024px to 4K px with GigaGAN? (Even if it takes a lot of memory and GPU/CPU)

In your code it seems to be a 256px upscale in 512px?

If it is possible to upscale in 1024px what values should I change in your code?

gan = GigaGAN(
    train_upsampler = True,     # set this to True
    generator = dict(
        style_network = dict(
            dim = 64,
            depth = 4
        ),
        dim = 32,
        image_size = 256,
        input_image_size = 64,
        unconditional = True
    ),
    discriminator = dict(
        dim_capacity = 16,
        dim_max = 512,
        image_size = 256,
        num_skip_layers_excite = 4,
        multiscale_input_resolutions = (128,),
        unconditional = True
    ),
    amp = True
).cuda()

dataset = ImageDataset(
    folder = '/path/to/your/data',
    image_size = 256
)

Should I change the dim, multiscale_input_resolutions? I'm a bit lost. :-)

Regards,

Sparfell