search

RaduAlexandru / permutohedral_encoding

Blazingly fast encoding for neural networks based on permutohedral lattices
MIT License
95 stars 10 forks source link

Multigpu question #3

Open Quyans opened 1 year ago

Quyans commented 1 year ago

Thanks for your great work. i test the code and it runs out here image

However, I found that if i set all device code from .device(torch::kCUDA, 0) to .device(torch::kCUDA, 1)) the encoded feature seems wrong image

do you know any solutions?