lkeab / gaussian-grouping

[ECCV'2024] Gaussian Grouping for open-world Anything reconstruction, segmentation and editing.
https://arxiv.org/abs/2312.00732
Apache License 2.0
500 stars 37 forks source link

Processing custom datasets with DEVA #33

Open leblond14u opened 2 months ago

leblond14u commented 2 months ago

Hi,

I'm trying to process a simple dataset with the DEVA module. However after only 2 images loaded I have the following error :

  2%|███                                                                                             | 2/101
RuntimeError: CUDA error: an illegal memory access was encountered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

AND 

  2%|███                                                                                             | 2/101
RuntimeWarning: Empty object mask!
RuntimeWarning: Trying to segment without any memory!

Have you ever encountered this issue ? Do you know how to solve it ?

Thanks a lot, Best regards,

Hugo