core.samplers#

Samplers for batching across sessions of unequal size.

DistributedSamplerWrapper

Wrapper for distributing any sampler across multiple processes.

SessionBatchSampler

Group any DatasetIndex sampler's output into per-session batches.