Searcher

class lightning_ir.retrieve.base.searcher.Searcher(index_dir: Path | str, search_config: SearchConfig, module: BiEncoderModule, use_gpu: bool = True)[source]

Bases: ABC

__init__(index_dir: Path | str, search_config: SearchConfig, module: BiEncoderModule, use_gpu: bool = True) → None[source]

Methods

`__init__`(index_dir, search_config, module[, ...])
`search`(output)
`to_gpu`()

Attributes

num_embeddings