Seeking GPU Memory Requirements for Model Training

[mohit-choithwani]

Hello, thanks for the fantastic work. I'm utilizing the NVIDIA GeForce RTX 3080 Lite Ha with a total of 64GB memory. I've experimented with various models and successfully executed a few like NeuS-Facto and BakedSDF. However, I encounter CUDA out-of-memory errors with other models. Could someone please provide the memory requirements for training each model? Thank you.

[niujinshuchong]

Hi, it depends on the model size and the batch size, number of samples. You can try different values.