How to reduce the GPU memory usage

fdarmon / NeuralWarp

Code release of paper Improving neural implicit surfaces geometry with patch warping

234 stars 14 forks source link

How to reduce the GPU memory usage #6

Closed BianFeiHu closed 2 years ago

BianFeiHu commented 2 years ago

Hello, I encountered OOM Error after "generate_visu" on RTX3080, can you provide some suggestions on how to reduce batch size or use multiple GPU?

fdarmon commented 2 years ago

Hello,

I have not implemented a multi-gpu version yet. You can decrease the number of rays to optimize from 512 to 400 with train/num_pixels=400 in NeuralWarp.conf. It should solve the OOM problem but the results may not look the same.