JonasSchult/Mask3D

Can the training be conducted on single 3090ti gpu?

Pixie8888 opened this issue · 1 comments

Pixie8888 commented

Hi,

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

JonasSchult commented

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

  • 👍1