sketchgraphs_models.distributed_utils.DistributedSampler

class sketchgraphs_models.distributed_utils.DistributedSampler(sampler, num_replicas=None, rank=None)

Utility class which adapts a sampler into a distributed sampler which only samples a subset of the underlying sampler, according to a division by rank.

__init__(sampler, num_replicas=None, rank=None)

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