FaissIVFIndexer

class lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer(index_dir: Path, index_config: FaissIVFIndexConfig, module: BiEncoderModule, verbose: bool = False)[source]

Bases: _FaissTrainIndexer

__init__(index_dir: Path, index_config: FaissIVFIndexConfig, module: BiEncoderModule, verbose: bool = False) None[source]

Methods

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

set_verbosity([verbose])

to_cpu()

to_gpu()

Attributes

INDEX_FACTORY