mixmo.loaders.batch_repetition_sampler.BatchRepetitionSampler

class mixmo.loaders.batch_repetition_sampler.BatchRepetitionSampler(sampler, batch_size, num_members, config_batch_sampler, drop_last=False)

Bases: torch.utils.data.sampler.BatchSampler

Wraps another sampler to yield a mini-batch of repeated indices.

__init__(sampler, batch_size, num_members, config_batch_sampler, drop_last=False)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(sampler, batch_size, num_members, …)	Initialize self.
output_format(std_batch)	Transforms standards batches into batches of sample summaries

output_format(std_batch)

Transforms standards batches into batches of sample summaries