search

JonasSchult / Mask3D

Mask3D predicts accurate 3D semantic instances achieving state-of-the-art on ScanNet, ScanNet200, S3DIS and STPLS3D.
MIT License
547 stars 110 forks source link

Training on multiple GPUs #60

Closed LinLin1031 closed 1 year ago

LinLin1031 commented 1 year ago

Hello!

I would like to ask a question. Is it possible to do your work on multiple GPUs at the same time?

I have three 16GB GPUs here and I would like to train on all three GPUs at the same time so that I can maybe achieve better results. If it is possible, can you give me some suggestions for modifying the code?

Best.

JonasSchult commented 1 year ago

Hi!

Right now, the codebase does not support multi-gpu training. It sounds like a great extension! :)

Best, Jonas