pytorch version

[IssamLaradji]

Great work!

Will a pytorch version be released for this code?

Thanks!

[chundiliu]

Hi Issam,

Thanks for being interested in our work! We do have a pytorch implementation, but it's not in shape yet. We will let you know once we clean everything up. But if you look into the code, most of the complicated operations are preprocessing and they are written in numpy. It won't be too hard to reuse them and rewrite the pytorch training part by yourself. Hope it helps.

Chundi