About the pytorch version.

[yurunsheng1]

Hi,

First of all, thank you so much for releasing this awesome code! I am deeply inspired by your paper.

But the tf version might be somewhat hard to read and I also find that you also willing to opensource the pytorch version. May I ask you whether you will open-source the pytorch version soon?

Thank you so much! Runsheng