TorchDenseSearcher

class lightning_ir.retrieve.pytorch.dense_searcher.TorchDenseSearcher(index_dir: Path, search_config: TorchDenseSearchConfig, module: BiEncoderModule, use_gpu: bool = True)

Bases: ExactSearcher

__init__(index_dir: Path, search_config: TorchDenseSearchConfig, module: BiEncoderModule, use_gpu: bool = True) → None

Methods

__init__(index_dir, search_config, module[, ...])
to_gpu()

Attributes