HELP! GPU memory overflow during testing.

[cvpr2023-3d-occupancy-prediction]

HELP! GPU memory overflow during testing.

[YWonchall]

Hi, What is the size of your GPU memory? Is it enough to train the baseline model?

[1349949]

fix it now.

[shenxiaowrj]

Hi! I used the bevformer_small_occ.py to train the model on my RTX 3090 (24G memory).
During the training process, the model is trained steadily. But when I was testing, when I got to the 3504th scene, the memory was beyond what my machine could handle.
But interestingly, my friend used two RTX 3090s to test the model, and his program was stopped when processing the 5900th scene because of memory issues.

[shenxiaowrj]

fix it now.

Thanks! I will try it now!