torch.cuda.OutOfMemoryError: CUDA out of memory.

JiehongLin / SAM-6D

[CVPR2024] Code for "SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation".

373 stars 32 forks source link

Open hanzibo opened 2 months ago

hanzibo commented 2 months ago

截图 2024-09-22 17-43-05 截图 2024-09-22 17-43-33 Can I change which profile to make it work

Goddard-Wen commented 1 week ago

请问你解决OOM这个问题了吗