probnmn.data.samplers¶

class probnmn.data.samplers.SupervisionWeightedRandomSampler(dataset: torch.utils.data.dataset.Dataset)[source]¶

Bases: torch.utils.data.sampler.WeightedRandomSampler

A torch.utils.data.WeightedRandomSampler to form a mini-batch with nearly equal number of examples with(out) program supervision during question coding and joint training.