DistributedSamplerWrapper

class stable_ssl.data.DistributedSamplerWrapper(sampler, *args, **kwargs)

Bases: DistributedSampler

Wrap a dataloader for DDP.

Parameters:

sampler (iterable) – The original dataset sampler.