torch.cuda.OutOfMemoryError: CUDA out of memory

NVlabs / InstantSplat

InstantSplat: Sparse-view SfM-free Gaussian Splatting in Seconds

https://instantsplat.github.io/

Other

861 stars 53 forks source link

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

Open bc129697 opened 3 months ago

bc129697 commented 3 months ago

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 commented 2 months ago

Hi, the OOM problem is caused by the memory limitations of DUSt3R: https://github.com/naver/dust3r/issues/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