search

Junyi42 / GeoAware-SC

Official Implementation of paper "Telling Left from Right: Identifying Geometry-Aware Semantic Correspondence"
https://telling-left-from-right.github.io
97 stars 8 forks source link

About memory leak #12

Open TLB-MISS opened 1 month ago

TLB-MISS commented 1 month ago

First of all, thank you for providing good code.

I think there is a memory leak in get_processed_features function in the demo. Each call consumes memory(about 3GB) far exceeding the return value(about 10MB). I'm guessing something keeps eating VRAM every time process_features_and_mask is called(Not sure), any ideas on how to fix this?

Thanks.