search

charlesq34 / pointnet

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Other
4.73k stars 1.45k forks source link

Which hardware specs should be enough to train pointNet? #88

Closed FilipaRamos closed 6 years ago

FilipaRamos commented 6 years ago

Hi! Thanks for your great work, I find PointNet imensely interesting.

I am wondering specially about the GPU and how much dedicated memory I should have in order to train pointnet on a custom dataset efficiently.

Thanks!

charlesq34 commented 6 years ago

HI @FilipaRamos

PointNet does not require very large memory for training, with batch size 32, pointnet_cls requires around 2.4G, pointnet_cls_basic requires around 1.3G. The segmentation models would require more memory though, but you can set smaller batch size to make it fit.

specifically we are using GTX 1080 and Titan Xp for our training.

FilipaRamos commented 6 years ago

Thanks @charlesq34 !