primeqa.ir.dense.colbert_top.colbert.trainer.Trainer

class primeqa.ir.dense.colbert_top.colbert.trainer.Trainer(triples, queries, collection, config=None)

Bases: object

Methods

best_checkpoint_path
configure
train	Note that config.checkpoint is ignored.

train(checkpoint='bert-base-uncased')

Note that config.checkpoint is ignored. Only the supplied checkpoint here is used.