search

zhulf0804 / PointPillars

A Simple PointPillars PyTorch Implementation for 3D LiDAR(KITTI) Detection.
MIT License
454 stars 112 forks source link

how to train with multi gpus!!! #46

Open lijh1024 opened 1 year ago

lijh1024 commented 1 year ago

I use torch.nn.DataParallel() to split the model, but it can not split the input as it is a list of tensor