search

wutong16 / Voxurf

[ ICLR 2023 Spotlight ] Pytorch implementation for "Voxurf: Voxel-based Efficient and Accurate Neural Surface Reconstruction"
Other
399 stars 28 forks source link

GPU out of memory #17

Closed pyni closed 1 year ago

pyni commented 1 year ago

Hi, Thanks for your code. I have met one problem. when I run the coarse stage, it works. But when I run the fine stage(even independently), it shows GPU out of memory :

image

I have tried to decrease the number of batch size (N_rand) in configs/default_fine_s.py, but it doesn't work.

Do you have any suggestions? Thanks.

wutong16 commented 1 year ago

Hi @pyni, sorry for the late reply. Decreasing the N_rand or using a lower resolution for the grid here would alleviate the problem, while the results may also be affected at the same time. The OOM problem usually occurs at around 15,000 iters during the fine stage, where the grid resolution is upscaled to 256^3.