google-research / omniglue

Code release for CVPR'24 submission 'OmniGlue'
https://hwjiang1510.github.io/OmniGlue
Apache License 2.0
439 stars 38 forks source link

OutOfMemoryError #11

Open lyxxxyl opened 4 weeks ago

lyxxxyl commented 4 weeks ago

Why does the model consume so much memory? I followed your suggestions to use CUDA acceleration, but during the demo run, my 6GB VRAM was insufficient, resulting in a torch.cuda.OutOfMemoryError. If I want to match two 700KB images, how much GPU memory would be required?

wujiazong commented 4 weeks ago

about 20GB image

lyxxxyl commented 2 days ago

about 20GB image

Thank you for your answer