Could you please provide an version compatible with PyTorch?

[ecnux]

Dear Author,
We would like to conduct experiments based on your code, but the version of TensorFlow you used is too early for us to use it effectively on our platform. When we attempted to reproduce the experiments using PyTorch, the results were not satisfactory. Could you please provide an updated version of the code or a version compatible with PyTorch?
Thank you.