search

MiguelMonteiro / permutohedral_lattice

Permutohedral Lattice C++/CUDA implementation + TensorFlow Op (CPU/GPU)
83 stars 18 forks source link

Pytorch? #18

Closed jgsimard closed 5 years ago

jgsimard commented 5 years ago

Hi, Just to know, will there be a PyTorch extension of this code? Thanks

MiguelMonteiro commented 5 years ago

Hello,

I have switched over to pytorch myself. Unfortunately, I don't have time at the moment to pursue this project further. But it is in the plans to transfer it over to pytorch in the future (just don't know when).

Best,

Miguel

jgsimard commented 5 years ago

Thanks for the quick reply!