FaissPQIndexConfig

class lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexConfig(num_train_embeddings: int | None = None, num_subquantizers: int = 16, n_bits: int = 8)

Bases: _FaissTrainIndexConfig

__init__(num_train_embeddings: int | None = None, num_subquantizers: int = 16, n_bits: int = 8) → None

Methods

__init__([num_train_embeddings, ...])
from_pretrained(index_dir)
save(index_dir)
to_dict()