search

sxyu / svox2

Plenoxels: Radiance Fields without Neural Networks
BSD 2-Clause "Simplified" License
2.79k stars 360 forks source link

CUDA out of memory #74

Closed RahulSinghalChicago closed 2 years ago

RahulSinghalChicago commented 2 years ago

Training on Colab Pro+ with 16GB GPU RAM and 54GB CPU, which should be enough for most models, but after a 6 epochs, the GPU runs out of memory. I suspect it's a caching issue and not an absolute memory issue. Does anyone know how to clear GPU in between epochs?

LiquidAmmonia commented 1 year ago

I come accross the same issue. May I ask how did you solve this problem?