search

matthewdm0816 / BridgeQA

[AAAI 24] Official Codebase for BridgeQA: Bridging the Gap between 2D and 3D Visual Question Answering: A Fusion Approach for 3D VQA
Other
15 stars 1 forks source link

Computing resource? #4

Closed codenewww closed 1 week ago

codenewww commented 2 months ago

Thanks for your code! Could you please specify the computational resources used for your experiments? Specifically, what type and number of GPUs were employed? If I only have one RTX 3090, can I finish the experiment?

matthewdm0816 commented 2 months ago

As in this issue, we train our model on 4 A100-80G GPUs for less than 10 hours (batch size=16). Supposedly, with smaller batch size, the training is doable but might take longer time on single 3090, e.g., 3-5 days