utils.samplers.multipack

utils.samplers.multipack

Multipack Batch Sampler

Classes

Name Description
MultipackBatchSampler Batch sampler class for multipack

MultipackBatchSampler

utils.samplers.multipack.MultipackBatchSampler(
    self,
    sampler,
    batch_size,
    batch_max_len,
    lengths,
    packing_efficiency_estimate=1.0,
    drop_last=False,
    num_count_samples=16,
    **kwargs,
)

Batch sampler class for multipack