DistributedProxySampler(sampler, num_replicas=None, rank=None)#
Distributed sampler proxy to adapt user’s sampler for distributed data parallelism configuration.
sampler – Input torch data sampler.
num_replicas – Number of processes participating in distributed training.
rank – Rank of the current process within
Input sampler is assumed to have a constant size.