Open IssamLaradji opened 4 years ago
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
Great work!
Will a pytorch version be released for this code?
Thanks!