分布式节点数据加载器
- class dgl.dataloading.DistNodeDataLoader(g, nids, graph_sampler, device=None, **kwargs)[source]
Bases:
DistDataLoader
用于分布式图存储的节点采样图数据加载器。
它在一组节点上封装了一个可迭代对象,生成消息流图(MFGs)列表,作为所述小批量在分布式图上的计算依赖。
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 – 参见
dgl.dataloading.DataLoader
.graph_sampler – See
dgl.dataloading.DataLoader
.device – See
dgl.dataloading.DataLoader
.kwargs – See
dgl.dataloading.DataLoader
.