primeqa.ir.dense.colbert_top.colbert.training.rerank_batcher.RerankBatcher

class primeqa.ir.dense.colbert_top.colbert.training.rerank_batcher.RerankBatcher(config: primeqa.ir.dense.colbert_top.colbert.infra.config.config.ColBERTConfig, triples, queries, collection, rank=0, nranks=1)

Bases: object

Methods

collate