Converting pth to Onnx models

I want to convert your pre-trained models to Onnx models to check for faster inferences, the [code](https://github.com/xinntao/Real-ESRGAN/blob/master/scripts/pytorch2onnx.py) given by @xinntao from [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) to convert their x4.pth model.
If I want to convert your models I suppose the same RDBnet was used, should I use the same code with your models or could you explain some changes that would make my work easier as I don't have .pb files for the same.