search

baaivision / Uni3D

[ICLR'24 Spotlight] Uni3D: 3D Visual Representation from BAAI
MIT License
497 stars 29 forks source link

About the gpu devices required for Evaluation of Zero-shot 3D classification #6

Open 11balanceNB26 opened 1 year ago

11balanceNB26 commented 1 year ago

Thank you for your great work. I am a newbie in this area and I would like to know what kind of GPU device is required when evaluating the model? I have two 3090 NVIDIA graphics cards, but this doesn't seem to be enough for evaluating the model. Is there any suitable method? Thanks! !

Wolfwjs commented 12 months ago

Thank you very much for your interest in our work! Regarding evaluating the model, we have not tested it on 3090 GPUs, but we have conducted successful single-card tests on both V100-32G and A100-40G GPUs. You could try reducing the batch size to see if that helps with your 3090 setup. Let us know if you encounter any further problems or have more questions!