mirrors
/
faiss
mirror of https://github.com/facebookresearch/faiss.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
faiss/contrib/torch
History
Maria Lomeli 9590ad2746 PQ with pytorch () 
Summary:
Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4116

This diff implements Product Quantization using Pytorch only.

Reviewed By: mdouze

Differential Revision: D67766798

fbshipit-source-id: fe2d44a674fc2056f7e2082e9765052c98fdc8f8
 		2025-01-06 09:48:32 -08:00
..
README.md
…
__init__.py
…
clustering.py PQ with pytorch () 2025-01-06 09:48:32 -08:00
quantization.py PQ with pytorch () 2025-01-06 09:48:32 -08:00

README.md

The Torch contrib

This contrib directory contains a few Pytorch routines that are useful for similarity search. They do not necessarily depend on Faiss.

The code is designed to work with CPU and GPU tensors.