Hello! Just wondering if there's a PyTorch implementation available?

[bo0018]

Hello!
I'm really interested in this project and was wondering if there's a PyTorch implementation available? If not, has anyone considered porting it to PyTorch?
Thanks for your time and keep up the great work！

[taojiashu]

This current version is written in PyTorch. Did you run into any problem in using the tool?