primeqa.mrc.trainers.mrc_mskd.IndividualDomainBatchSampler

class primeqa.mrc.trainers.mrc_mskd.IndividualDomainBatchSampler(dataset, batch_size)

Bases: torch.utils.data.sampler.Sampler

iterate over domains and provide a batch per domain in each mini-batch

Methods