Luoyadan / KECOR-active-3Ddet

The official PyTorch implementation of "KECOR: Kernel Coding Rate Maximization for Active 3D Object Detection" (ICCV 2023).
Apache License 2.0
8 stars 1 forks source link

Dist train #5

Open AlanLiangC opened 2 days ago

AlanLiangC commented 2 days ago

HI, Prof. Luo. Thanks for your excelent waork. Does the training process only work for a single GPU? As you can see, I got many errors when train with multi-GPU.

Luoyadan commented 1 day ago

Hi Alan,

Yes, we tested the code using only a single GPU. You may need to modify the code to ensure sample selection works with rank=0.

Cheers, Yadan