ForMyCat / SparseGS

Codebase for SparseGS paper
Other
103 stars 11 forks source link

about OOM #5

Closed YZWYD closed 4 months ago

YZWYD commented 4 months ago

Hello, thank you very much for open-sourcing this work. I encountered an out-of-memory issue when trying to reproduce it, as my device only has 8GB of CUDA memory. What should I modify to avoid this from happening?

ForMyCat commented 4 months ago

Try playing with the -r argument. (i.e. -r 8 will scale all the training images to 1/8 resolution) You can also try disabling SDS and Warping. 8G is a bit tight. I recommend you start with a lower resolution and add one component at a time to test. (depth Pearson loss --> floater pruning --> SDS --> Warping)

YZWYD commented 4 months ago

Thank you so much for your prompt reply! By the way, when you mentioned SDS and Warping in your response, are you referring to Diffusion Params and Depth Warping Loss parms? (I apologize for any ambiguity in my understanding due to my limited English proficiency.)

ForMyCat commented 4 months ago

Yes, sorry for the confusion. SDS uses Stable Diffusion as backbone and Depth Warping Loss can be viewed as a regularization term, hence the naming.

YZWYD commented 4 months ago

Thank you once again for your prompt response! As a newcomer to the field of 3D vision, I am still becoming familiar with some of the terminology, which may have caused some communication barriers. I apologize for any inconvenience.

ForMyCat commented 4 months ago

Don't worries! Thank you for your interest in our project! Feel free to open a new issue if you have further questions.