gigagan-pytorch
gigagan-pytorch copied to clipboard
[Question] About the upscaler
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