search

hxdengBerkeley / PointCNN.Pytorch

A PyTorch implemetation of PointCNN (CS294-131 Course Proj)
MIT License
219 stars 37 forks source link

Slow training #1

Closed lynetcha closed 6 years ago

lynetcha commented 6 years ago

Thanks for the great implementation! I notice training is quite slow. Any pointers on what part of the code can be optimized for speed?

lynetcha commented 6 years ago

Nevermind. I found the note about KNN. Thanks!

iSarmad commented 6 years ago

hello, can you suggest how to improve training time?, it takes excruciatingly long at the moment uses only 10 to 15 % of GPU resources, thanks