How to train ESRGAN.tflite model for custom input?

Hello,

I’m working on an Image Super-Resolution Model called ESRGAN.tflite. Currently, it takes a 50X50 size image as an input and a 200X200 size as the output. But I want,

It accepts 240X240 as input and generates 960X960 as output.

Please Help me to achieve this.