And I also facing the same problem with ESRGAN (Image super-resolution) model, the model accepts 50X50 image size and outputs 200X200.
My question is how to train the model for custom input, like 150X150 or 240X240.
here is the link.

  
      

      TensorFlow
  

  
    

Super resolution with TensorFlow Lite 1