Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

torch.cuda.OutOfMemoryError: CUDA out of memory #10

Open
bc129697 opened this issue Aug 16, 2024 · 2 comments
Open

torch.cuda.OutOfMemoryError: CUDA out of memory #10

bc129697 opened this issue Aug 16, 2024 · 2 comments

Comments

@bc129697
Copy link

image

When i run with "bash scripts/run_train_infer.sh" ,This error will occur when using 60 frames of images, but it will not occur when using 25 frames of images

@kairunwen
Copy link

kairunwen commented Aug 27, 2024

Hi, the OOM problem is caused by the memory limitations of DUSt3R: naver/dust3r#1, which should be solved by InstantSplat_MASt3R.

You can modify these parameters to reduce GPU memory usage and accelerate model training, but the model performance will be slightly degraded: https://github.com/NVlabs/InstantSplat/blob/main/coarse_init_infer.py#L70
(1) scene_graph='complete' --> scene_graph='swin'
(2) symmetrize=True --> symmetrize=False

@alpergel
Copy link

alpergel commented Dec 6, 2024

Hi, where can we find InstantSplat_MAST3R?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants