nerfbaselines icon indicating copy to clipboard operation
nerfbaselines copied to clipboard

Memory problems with custom dataset

Open Mikey-E opened this issue 5 months ago • 2 comments

For gaussian-opacity-fields it appears that grads and grads_abs may not be getting cleared correctly between calls of densify_and_prune(). They seem to double in size each time, eventually leading to out of memory errors. As such, simply changing memory parameters or environment variables does little to help. What is the best way to manage this size during training?

Mikey-E avatar Sep 24 '24 06:09 Mikey-E