Using trained model for custom sized images

[malalejandra]

Hi @alexlee-gk. Thank you for your wonderful code, was quite an interesting journey to dig into it!

I trained models for my own dataset of 64x64 images and wondering if it's possible to use the model to sample 512x512 images? If yes, what changes I should make so that the model become scalable?

Thank you!

[blackPython]

To run for 512x512 image you have to change the enocder and decoder.. Not the architecture but the number layers or step size to account for more number of pixels.