primeqa.ir.dense.colbert_top.colbert.training.eager_batcher_v2.EagerBatcher

class primeqa.ir.dense.colbert_top.colbert.training.eager_batcher_v2.EagerBatcher(config: primeqa.ir.dense.colbert_top.colbert.infra.config.config.ColBERTConfig, triples, rank=0, nranks=1, is_teacher=False)

Bases: object

Methods

collate
shuffle
skip_to_batch