Memory cost went too high

3dlg-hcvc / M3DRef-CLIP

[ICCV 2023] Multi3DRefer: Grounding Text Description to Multiple 3D Objects

https://3dlg-hcvc.github.io/multi3drefer/

MIT License

64 stars 3 forks source link

Memory cost went too high #12

Closed Victor-Xiaoyu-Ye closed 8 months ago

Victor-Xiaoyu-Ye commented 8 months ago

I have added number of views, and the computation cost went bigger than 24GB as i'm using a single 3090.

Is there any way I can reduce the computation cost while the performance won't down drasticly?

Many thanks.

eamonn-zh commented 8 months ago

Hi @victor-cilay,

How many views are you currently using, is it the only change you've made?

Victor-Xiaoyu-Ye commented 8 months ago

12, yes it's the only change

eamonn-zh commented 8 months ago

It is not surprising you run out of the memory with 12 views. You can try reducing the batch size and feeding smaller batch of images to the renderer. But we haven't tested such settings.