GFPGAN model - train for 256x256

Being given the GFPGAN model, I want to speed up the inference by training the model for 25x256 images. (currently it is for 512x512)

https://github.com/TencentARC/GFPGAN

I will provide access to AWS EC2(with GPU) instance for training.

Python/ML/PyTorch gurus are welcome to apply for this job.