FaissPQIndexer

class lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexer(index_dir: Path, index_config: FaissPQIndexConfig, module: BiEncoderModule, verbose: bool = False)

Bases: _FaissTrainIndexer

__init__(index_dir: Path, index_config: FaissPQIndexConfig, module: BiEncoderModule, verbose: bool = False) → None

Methods

__init__(index_dir, index_config, module[, ...])
to_cpu()
to_gpu()

Attributes

INDEX_FACTORY