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

NVIDIAGameWorks / kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

Apache License 2.0

4.32k stars 537 forks source link

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

Open sparse-mvs-2 opened 2 months ago

sparse-mvs-2 commented 2 months ago

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?

Caenorst commented 2 months ago

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