kaolin icon indicating copy to clipboard operation
kaolin copied to clipboard

Asking support for multiple-gpu training of structed point cloud class

Open sparse-mvs-2 opened this issue 1 year ago • 2 comments

I want to use nn.DataParallel for multiple-gpu training of sparse convolution in structed point cloud. However, the spc class contain spc.octrees, which is of data type uint8. So It will say "Unconvertible NCCL type Short". This is bad if I want to scale the network size up. Currently I have no solution to this problem. Is there any suggestions?

sparse-mvs-2 avatar Apr 24 '24 08:04 sparse-mvs-2

Actually from this issue, it looks like the issue is from the int16. I'll look into it.

Caenorst avatar May 01 '24 17:05 Caenorst