Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

compute_jaccard_distance function failing on multi-gpu #9

Open
toshalpatel opened this issue Sep 29, 2022 · 0 comments
Open

compute_jaccard_distance function failing on multi-gpu #9

toshalpatel opened this issue Sep 29, 2022 · 0 comments

Comments

@toshalpatel
Copy link

toshalpatel commented Sep 29, 2022

Using compute_jaccard_distance function defined in faiss_rerank.py. The search_option variable is hard coded to 2, which uses multiple GPUs if the machine has any.

The final jaccard distance is incorrect as each value is the same (some number), whereas when i use search_option = None, i.e., CPU option, the jaccard distance makes sense.

I am using RTX 3090 GPU.

Can you please help me check if there is some issue?
Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant