search

JonasSchult / Mask3D

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

Can the training be conducted on single 3090ti gpu? #127

Closed Pixie8888 closed 1 year ago

Pixie8888 commented 1 year ago

Hi,

Thanks for releasing code. I am wondering if the training can be conducted on a 3090ti (24G memory)?

JonasSchult commented 1 year ago

Hi!

Networks trained on 2cm voxelization unfortunately need more than 24GB RAM. You can set data.voxel_size=0.05 but this will result in worse performance.

Best, Jonas