image size problem

[Zwhy0712]

hello ！Thank you for developing a pytorch version of comodgan！
I would like to ask if the size of this image can only be a square or an arbitrary scale

[ayush12gupta]

in this implementation it can just be square