DistNodeDataLoaderο
- class dgl.dataloading.DistNodeDataLoader(g, nids, graph_sampler, device=None, **kwargs)[source]ο
Bases:
DistDataLoader
Sampled graph data loader over nodes for distributed graph storage.
It wraps an iterable over a set of nodes, generating the list of message flow graphs (MFGs) as computation dependency of the said minibatch, on a distributed graph.
All the arguments have the same meaning as the single-machine counterpart
dgl.dataloading.DataLoader
except the first argumentg
which must be adgl.distributed.DistGraph
.- Parameters:
g (DistGraph) β The distributed graph.
nids β See
dgl.dataloading.DataLoader
.graph_sampler β See
dgl.dataloading.DataLoader
.device β See
dgl.dataloading.DataLoader
.kwargs β See
dgl.dataloading.DataLoader
.
See also