primeqa.ir.dense.colbert_top.colbert.training.lazy_batcher.LazyBatcher

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

Bases: object

Methods

collate
shuffle
skip_to_batch