Skip to content

Instantly share code, notes, and snippets.

@mdouze

mdouze/demo_two_level_clustering.ipynb Secret

Last active October 26, 2023 09:55
Show Gist options
  • Save mdouze/1b2483d72c0b8984dd152cd81354b7b4 to your computer and use it in GitHub Desktop.
Save mdouze/1b2483d72c0b8984dd152cd81354b7b4 to your computer and use it in GitHub Desktop.
Download ZIP
Display the source blob
Display the rendered blob
Raw
demo_two_level_clustering.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@mdouze
Copy link
Author

mdouze commented Mar 9, 2023

Sure, just pass gpu=True as an additional argument to train_ivf_index_with_2level

@hustnn
Copy link

hustnn commented Oct 26, 2023

hi, @mdouze Can I use the same idea for index IVF262144(IVF512,PQ64x4fs,RFlat),PQ64 in c++? The training in cpu is too slow.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment