-
Notifications
You must be signed in to change notification settings - Fork 3
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
Enable torch #35
Enable torch #35
Conversation
Since it seems to be more or less impossible to provide type hints for a library (torch), that may or may not be available at runtime, I've given up and used On the upside: All |
Since Faiss has torch support with tensors that are on the GPU, it is desirable to avoid moving these to and from the GPU.
This PR adds the ability to have an entire pipeline for hubness reduced nearest neighbor search on the GPU.
For some hubness reduced methods it is not feasible to do this (probably only MP-empiric).
TODO: