Replies: 4 comments 12 replies
-
Thanks in advance for the discussion. Do you have a small example to showcase what you mean exactly?
|
Beta Was this translation helpful? Give feedback.
-
Just realized that
|
Beta Was this translation helpful? Give feedback.
-
Is it possible to pass a DistributedSampler to NeighborLoader as it is the case for torch.utils.data.DataLoader ? When I tried to do that the batches only contain a few elements of the original data. |
Beta Was this translation helpful? Give feedback.
-
Hi! Additionally, I have a question regarding the testing process. Did they use the entire dataset for testing? If so, won't that also lead to a CUDA out of memory issue for large datasets like OGBN-Products? https://github.com/pyg-team/pytorch_geometric/blob/master/examples/reddit.py |
Beta Was this translation helpful? Give feedback.
-
🚀 Feature
Add a sampler argument to NeighborLoader. The sampler would return a list of nodes to load.
Motivation
Tow keys motivations.
Beta Was this translation helpful? Give feedback.
All reactions