search

quiver-team / torch-quiver

PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.
https://torch-quiver.readthedocs.io/en/latest/
Apache License 2.0
293 stars 36 forks source link

fix sampler #91

Closed eedalong closed 3 years ago

eedalong commented 3 years ago

Fix sampler when size > neighbor_num

eedalong commented 3 years ago

@lausannel @ZenoTan please merge this branch with your own branch and test with it