Hi,
Is the value of num_hops
in calling NeighborSampler
the same as the value of NodeFlow.num_layers
that it returns?
There are two different ways of doing this
You are right when you want to use FullNeighborSampler
sampler = dgl.dataloading.MultiLayerFullNeighborSampler(nof_layers)
You can also set the number of neighbors you want to sample for each layer like below
sampler = dgl.dataloading.MultiLayerNeighborSampler([15, 10]) #for 2 layers
See the documentation for more details: dgl.dataloading — DGL 0.6.1 documentation
1 Like