How many GPUs are needed to train your model

OpenMask3D / openmask3d

MIT License

184 stars 13 forks source link

How many GPUs are needed to train your model #8

Closed xmu-xiaoma666 closed 10 months ago

xmu-xiaoma666 commented 10 months ago

How many GPUs are needed to train your model?

aycatakmaz commented 10 months ago

Please refer to Mask3D repository for details about the training requirements of the mask proposal module. The model we used was trained with a single GPU. For all other parts of the pipeline (running SAM and CLIP), we used a single GPU.